View source for 0598 - Gears

= Cerința =
Considerăm un ansamblu format din <code>n</code> roți dințate, numerotate de la <code>1</code> la <code>n</code>, 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 <code>n</code>, numărul de ordine <code>x</code> 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 <code>gearsIN.txt</code> conține pe prima linie numerele <code>n x</code>. Următoarele <code>n-1</code> linii vor conține perechi de numere <code>i j</code> reprezentând perechi de roți conectate.

= Date de ieșire =
Fișierul de ieșire <code>gearsOUT.txt</code> va conține pe prima linie un șir de <code>n</code> caractere: <code>S</code> sau <code>D</code>, 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 =

* <code>1 ≤ n ≤ 100</code>

= Exemplul 1: =
<code>gearsIN.txt</code>
 7 4
 1 2
 1 3
 1 4
 1 5
 3 6
 3 7
<code>gearsOUT.txt</code>
 SDDDDSS

= Exemplul 2: =
<code>gearsIN.txt</code>
 1001 23
 1 2
 1 3
 1 4
 1 5
 3 6
 3 7
<code>gearsOUT.txt</code>
 Datele nu corespund restrictiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
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")

</syntaxhighlight>