Vizualizare sursă pentru „0650 – kNivel”

== 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.

= Cerința =
Se dă vectorul de tați al unui arbore cu rădăcină cu <code>n</code> noduri și o valoare <code>k</code>. Determinați nodurile situate pe nivelul <code>k</code> în arbore.

= Date de intrare =
Fișierul de intrare <code>knivelIN.txt</code> conține pe prima linie numărul de noduri <code>n</code> și valoarea <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>knivelOUT.txt</code> va conține pe prima linie, în ordine crescătoare, nodurile situate pe nivelul <code>k</code> în arbore.Î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>
* pe nivelul <code>k</code> va exista cel puțin un nod

= Exemplul 1: =
<code>knivelIN.txt</code>
 8 3
 4 3 0 3 2 1 2 1
<code>knivelOUT.txt</code>
 1 5 7

= Exemplul 2: =
<code>knivelIN.txt</code>
 101 2
 4 3 0 3 2 1 2 1
<code>knivelOUT.txt</code>
 Datele nu corespund restrictiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def verificare_restrictii(n, k):
    if 1 <= n <= 100:
        return True
    else:
        return False

def gaseste_noduri_pe_nivel(tati, nivel):
    noduri_pe_nivel = []
    for i in range(len(tati)):
        nivel_nod_curent = 1  # Nivelul rădăcinii
        parinte_curent = tati[i]

        while parinte_curent != 0:
            parinte_curent = tati[parinte_curent - 1]
            nivel_nod_curent += 1

        if nivel_nod_curent == nivel:
            noduri_pe_nivel.append(i + 1)

    return noduri_pe_nivel

def main():
    # Citirea datelor de intrare din fișier
    with open("knivelIN.txt", "r") as fisier_intrare:
        n, k = map(int, fisier_intrare.readline().split())
        tati = list(map(int, fisier_intrare.readline().split()))

    # Verificarea restricțiilor
    if not verificare_restrictii(n, k):
        with open("knivelOUT.txt", "w") as fisier_iesire:
            fisier_iesire.write("Datele nu corespund restrictiilor impuse")
        return

    # Găsirea nodurilor de pe nivelul k
    noduri_pe_nivel_k = gaseste_noduri_pe_nivel(tati, k)

    # Scrierea rezultatului în fișierul de ieșire
    with open("knivelOUT.txt", "w") as fisier_iesire:
        if noduri_pe_nivel_k:
            fisier_iesire.write(" ".join(map(str, noduri_pe_nivel_k)))
        else:
            fisier_iesire.write("Nu există noduri pe nivelul specificat")

if __name__ == "__main__":
    main()

</syntaxhighlight>