Vizualizare sursă pentru „0643 – Det Drum1”

= Cerința =
Se dă vectorul de tați al unui arbore cu rădăcină cu <code>n</code> noduri și un nod <code>k</code>. Determinați drumul de la rădăcina arborelui la nodul <code>k</code>.

= Date de intrare =
Fișierul de intrare <code>detdrum1IN.txt</code> conține pe prima linie numărul de noduri <code>n</code> și nodul <code>k</code>. Pe linia următoare se află vectorul de tați al arborelui, valorile fiind separate prin spații.

= Date de ieșire =
Fișierul de ieșire <code>detdrum1OUT.txt</code> va conține pe prima linie nodurile care alcătuiesc drumul determinat, separate printr-un spațiu.

= Restricții și precizări =

* <code>1 ≤ n ≤ 100</code>
* <code>1 ≤ k ≤ n</code>
* în vectorul de tați rădăcina este marcată cu <code>0</code>

== Exemplul 1 ==
detdrum1IN.txt:

7 2

4 1 7 0 7 7 1

detdrum1OUT.txt:

4 1 2

== Exemplul 2 ==
detdrum1IN.txt:

101 2

4 1 7 0 7 7 1

Output:

Invalid input. Please check the conditions.

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def is_valid_input(n, k):
    return 1 <= n <= 100 and 1 <= k <= n

with open("detdrum1IN.txt", "r") as infile, open("detdrum1OUT.txt", "w") as outfile:
    n, k = map(int, infile.readline().split())

    if not is_valid_input(n, k):
        print("Invalid input. Please check the conditions.")
    else:
        t = [0] + list(map(int, infile.readline().split()))

        def detdrum(nod):
            if nod == 0:
                return
            detdrum(t[nod])
            outfile.write(str(nod) + " ")

        detdrum(k)

</syntaxhighlight>