0598 - Gears

Cerința

Considerăm un ansamblu format din n roți dințate, numerotate de la 1 la n, ca în imaginea alăturată. Fiecare roată se poate roti spre dreapta sau spre stânga. Dacă o roată se rotește spre dreapta, toate roțile pe care le angrenează se vor roti spre stânga, și invers.

Una dintre roți este conectată la un motor și se va roti spre dreapta, iar toate roțile din ansamblu se vor roti în mod corespunzător. Ansamblul este construit astfel încât toate roțile vor fi angrenate și fiecare roată va fi angrenată de o unică altă roată.

Dându-se numărul de roți n, numărul de ordine x al roții conectate la motor și perechile de roți conectate între ele, să se determine sensul de rotație al fiecărei roți.

Date de intrare

Fișierul de intrare gearsIN.txt conține pe prima linie numerele n x. Următoarele n-1 linii vor conține perechi de numere i j reprezentând perechi de roți conectate.

Date de ieșire

Fișierul de ieșire gearsOUT.txt va conține pe prima linie un șir de n caractere: S sau D, după cum roata corespunzătoare se rotește spre stânga, respectiv dreapta. În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Nu corespunde restricțiilor impuse".

Restricții și precizări

• 1 ≤ n ≤ 100

Exemplul 1:

gearsIN.txt

7 4
1 2
1 3
1 4
1 5
3 6
3 7

gearsOUT.txt

SDDDDSS

Exemplul 2:

gearsIN.txt

1001 23
1 2
1 3
1 4
1 5
3 6
3 7

gearsOUT.txt

Datele nu corespund restrictiilor impuse

Rezolvare

def verifica_restrictii(n):
    if 1 <= n <= 100:
        return True
    else:
        return False

def citeste_intrare(nume_fisier):
    with open(nume_fisier, 'r') as fisier:
        n, x = map(int, fisier.readline().split())
        conexiuni = [tuple(map(int, linie.split())) for linie in fisier.readlines()]
    return n, x, conexiuni

def scrie_iesire(nume_fisier, rezultat):
    with open(nume_fisier, 'w') as fisier:
        fisier.write(rezultat)

def directii_rotire(n, x, conexiuni):
    graf = {i: {'directie': 'S'} for i in range(1, n+1)}
    graf[x]['directie'] = 'D'

    for conexiune in conexiuni:
        i, j = conexiune
        if graf[i]['directie'] == 'S':
            graf[j]['directie'] = 'D'
        else:
            graf[j]['directie'] = 'S'

    return ''.join(graf[i]['directie'] for i in range(1, n+1))

if __name__ == "__main__":
    fisier_intrare = "gearsIN.txt"
    fisier_iesire = "gearsOUT.txt"

    n, x, conexiuni = citeste_intrare(fisier_intrare)

    if verifica_restrictii(n):
        rezultat = directii_rotire(n, x, conexiuni)
        scrie_iesire(fisier_iesire, rezultat)
    else:
        scrie_iesire(fisier_iesire, "Datele nu corespund restrictiilor impuse")