Editing 0645 – Det Drum3

= Cerința =
Se dă vectorul de tați al unui arbore cu rădăcină cu <code>n</code> noduri în care fiecare nod are asociată o valoare numerică. Determinați drumul de la rădăcină la un nod terminal pentru care suma valorilor asociate nodurilor este maximă.

= Date de intrare =
Fișierul de intrare <code>detdrum3IN.txt</code> conține pe prima linie numărul de noduri <code>n</code>. Pe linia următoare se află vectorul de tați al arborelui, valorile fiind separate prin spații. Pe următoarea linie se află, în ordine, valorile asociate nodurilor din arbore.

= Date de ieșire =
Fișierul de ieșire <code>detdrum3OUT.txt</code> va conține pe prima linie suma maxima determinată, iar pe a doua linie nodurile care alcătuiesc drumul determinat, separate printr-un spațiu. Dacă există mai multe noduri terminale pentru care suma valorilor nodurilor situate pe drumul de la rădăcină la nod este maximă, se va afișa drumul spre nodul cu numărul de ordine minim.

= Restricții și precizări =

* <code>1 ≤ n ≤ 100</code>
* în vectorul de tați rădăcina este marcată cu <code>0</code>
* valorile asociate nodurile din arbore sunt numere naturale mai mici decât <code>1000</code>

== Exemplul 1 ==
detdrum3IN.txt:

8

4 3 0 3 2 1 4 1

2 5 2 1 3 5 3 1

detdtrum3OUT.txt:

10

3 2 5 

== Exemplul 2 ==
detdrum3IN.txt:

101

4 3 0 3 2 1 4 1

2 5 2 1 3 5 3 1

Output:

Input necorespunzator

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def verificare(n, v):
    if 1>n or 101<=n:
        return False
    for i in range(len(v)):
        if v[i] < 0 or v[i] > 1000:
            return False
    return True

with open("detdrum3IN.txt") as f:
    n = int(next(f))
    t = [0] + list(map(int, next(f).split()))
    v = [0] + list(map(int, next(f).split()))
    if not verificare(n,v):
        print("Input necorespunzator")
        exit()

result = []
maxim = float("-inf")
for i in range(1, n + 1):
    c = 0

    for x in t[1:]:
        if x == i:
            c += 1

    if c == 0:
        p = t[i]
        s = v[i]

        while p:
            s += v[p]
            p = t[p]

        maxim = max(maxim, s)

with open("detdrum3OUT.txt", "w") as fout:
    print(maxim, file=fout)

    valid = False

    for i in range(1, n + 1):
        c = 0

        for x in t[1:]:
            if x == i:
                c += 1

        if c == 0:
            p = t[i]
            s = v[i]

            while p:
                s += v[p]
                p = t[p]

            if s == maxim and not valid:
                valid = True
                result.append(i)
                p = t[i]
                while p:
                    result.append(p)
                    p = t[p]

    for x in result[::-1]:
        print(x, end=" ", file=fout)
</syntaxhighlight>