Editing 0639 – Înălțime

== Enunț ==
Într-un arbore cu rădăcină, spunem că rădăcina este pe nivelul <code>1</code>, fiii rădăcinii pe nivelul <code>2</code>, fiii fiilor rădăcinii pe nivelul <code>3</code>, etc. Numărul de nivele distincte din arbore determină înălțimea arborelui.

= Cerința =
Se dă vectorul de tați al unui arbore cu rădăcină cu <code>n</code> noduri. Determinați înălțimea arborelui.

= Date de intrare =
Fișierul de intrare <code>inaltimeIN.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.

= Date de ieșire =
Fișierul de ieșire <code>inaltimeOUT.txt</code> va conține pe prima linie numărul <code>H</code> , reprezentând înălțimea arborelui.Î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>
* în vectorul de tați rădăcina este marcată cu <code>0</code>

= Exemplu 1: =
<code>inaltimeIN.txt</code>
 7
 4 1 7 0 7 7 1 
<code>inaltimeOUT.txt</code>
 4

= Exemplu 2: =
<code>inaltimeIN.txt</code>
 101
 4 1 7 0 7 7 1 
<code>inaltimeOUT.txt</code>
 Datele nu corespund restrictiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def verifica_restrictii(n):
    if 1 <= n <= 100:
        return True
    else:
        with open("inaltimeOUT.txt", "w") as file:
            file.write("Datele nu corespund restrictiilor impuse")
        return False

def calculeaza_inaltimea(n, tati):
    if not verifica_restrictii(n):
        return

    inaltime = 0
    nivele = [0] * n

    for i in range(n):
        nivel = 0
        j = i
        while tati[j] != 0:
            nivel += 1
            j = tati[j] - 1
            if nivele[j] != 0:
                nivel += nivele[j]
                break

        nivele[i] = nivel
        inaltime = max(inaltime, nivel + 1)

    with open("inaltimeOUT.txt", "w") as file:
        file.write(str(inaltime))

def main():
    # Citirea datelor de intrare
    with open("inaltimeIN.txt", "r") as file:
        n = int(file.readline())
        if not verifica_restrictii(n):
            return
        tati = list(map(int, file.readline().split()))

    # Calcularea inaltimii arborelui
    calculeaza_inaltimea(n, tati)

if __name__ == "__main__":
    main()

</syntaxhighlight>