3565 - Furnici 1

La Institutul de cercetare al insectelor s-a descoperit că dacă furnicile sunt puse pe o bară metalică, ele au un comportament bine definit după următoarele reguli:
1. Imediat cum a fost pusă pe bară ea îşi începe deplasarea în sensul în care a fost orientată, cu viteza constantă de 1 cm/s. Furnica nu se opreşte cât timp se află pe bara metalică chiar dacă se ciocneşte cu altă furnică.
2. Dacă pe drum nu se întâlneşte cu altă furnică ea îşi va continua deplasarea până când va cădea de pe bară.
3. Când două furnici se întâlnesc, ele îşi schimbă amândouă instantaneu sensul de deplasare.

Cerinţa

Ştiind că pe o bară metalică de lungime L cm se plasează exact N furnici în poziţii cunoscute şi cu sensul iniţial de deplasare cunoscut, să se scrie un program care calculează numărul de secunde după care va cădea de pe bară şi ultima furnică de la momentul iniţial. Toate furnicile îşi încep deplasarea concomitent.

Date de intrare

Fişierul de intrare furnici1in.txt conţine pe prima linie două numere naturale L şi N separate printr-un spaţiu. Apoi urmează N linii cu câte 2 valori: pozi şi sensi separate printr-un spaţiu, pozi este un număr natural care reprezintă coordonata la care se află furnica i la momentul iniţial, iar sensi este un caracter din mulţimea {‘S’,’D’} ce arată sensul de deplasare iniţial pe care îl are furnica i (S pentru stânga şi D pentru dreapta).

Date de ieşire

Fişierul de ieşire furnici1out.txt va conţine un singur număr care reprezintă timpul la care a căzut ultima furnică.

Restricții și precizări

• 1 < L < 10.000.000
• 0 < N < 100.000
• 0 ≤ pozi ≤ L

Exemplul 1

furnici1in.txt

10 3

4 D

8 S

1 S

furnici1out.txt

Datele introduse corespund restricțiilor impuse.

8

Exemplul 2

furnici1in.txt

10 3

4 D

8 S

1 10000

furnici1out.txt

Datele introduse nu corespund restricțiilor impuse.

Explicație

Bara are lungimea de 10 cm şi pe bară vor fi plasate 3 furnici la distanţele de 4, 8, 1 faţă de capătul din stânga al barei şi vor avea următoarele sensuri de deplasare: dreapta, stânga respectiv stânga. Primele două furnici se vor întâlni după 2 secunde în punctul de coordonată 6 şi îşi vor schimba sensul de deplasare. După schimbarea de sens a doua furnică va parcurge încă 4 cm către dreapta şi va cădea, în timp ce prima furnică după schimbarea de sens va parcurge încă 6 cm către stânga, până la cădere. În acest timp a treia furnică a căzut de pe bara după 1 secundă. Pe bară nu vor mai fi furnici după 8 secunde.

Rezolvare

# 3565 - Furnici 1
def validare(lungime1, furnici):           # functia de validare a datelor de intrare
    if lungime1 < 1 or lungime1 > 10_000_000:
        raise ValueError

    if len(furnici) < 0 or len(furnici) > 100_000:
        raise ValueError

    for pozi, directie in furnici:
        if pozi < 0 or pozi > lungime1:
            raise ValueError
        if directie not in ['S', 'D']:
            raise ValueError

    file_out.write("Datele de intrare corespund restrictiilor impuse\n")


def last_ant_falls(lungime1, furnici):                     # functia de rezolvare
    max_time = 0
    for pozi, directie in furnici:
        if directie == 'S':
            time = pozi
        else:  # direction == 'D'
            time = lungime1 - pozi
        max_time = max(max_time, time)
    return max_time


if __name__ == '__main__':
    file_in = open("furnici1in.txt", "r")         # declararea fisierelor
    file_out = open("furnici1out.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

    # din cauza datelor de intrare pot aparea 2 tipuri de erori, valueError sau IndexError pe care le tratam
    try:
        lungime, n = map(int, file_in.readline().split())
        ants = []
        for _ in range(n):
            pos, direction = file_in.readline().split()
            pos = int(pos)
            ants.append((pos, direction))

        validare(lungime, ants)                 # apelul functiei de validare
        result = last_ant_falls(lungime, ants)  # apelul functiei de rezolvare
        file_out.write(str(result))

    except ValueError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")
    except IndexError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")