Vizualizare sursă pentru „4156 – Nivele Pare”

== Enunț ==
Într-un arbore cu rădăcină, nivelul unui nod este lungimea lanțului de la rădăcină la acel nod. Astfel, rădăcina este pe nivelul <code>0</code>, fiii rădăcinii pe nivelul <code>1</code>, fiii fiilor rădăcinii pe nivelul <code>2</code>, etc.

= Cerința =
Se dă vectorul de tați al unui arbore cu rădăcină cu <code>n</code> noduri. Determinați nodurile situate pe nivele pare.

= Date de intrare =
Fișierul de intrare <code>nivelepareIN.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>nivelepareOUT.txt</code> va conține mai multe linii. Fiecare linie <code>i</code> conține, în ordine crescătoare, nodurile aflate pe nivelul <code>2*(i-1)</code>.Î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>

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

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

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

def citeste_arbore_din_fisier(file_path):
    with open(file_path, 'r') as file:
        n = int(file.readline())
        if not verifica_restrictii(n):
            return None, None, "Datele nu corespund restrictiilor impuse"
        parinti = list(map(int, file.readline().split()))
    return n, parinti, None

def scrie_noduri_pe_nivele_pare(file_path, nivele_pare):
    with open(file_path, 'w') as file:
        for nivel in nivele_pare:
            file.write(' '.join(map(str, nivel)) + '\n')

def noduri_pe_nivele_pare(n, parinti):
    nivele_pare = [[] for _ in range(n)]

    for i in range(n):
        parinte = parinti[i]
        nivel = 0
        while parinte != 0:
            nivel += 1
            parinte = parinti[parinte - 1]

        if nivel % 2 == 0:
            nivele_pare[nivel // 2].append(i + 1)

    return nivele_pare

if __name__ == "__main__":
    input_file = "nivelepareIN.txt"
    output_file = "nivelepareOUT.txt"

    n, parinti, restrictii_msg = citeste_arbore_din_fisier(input_file)

    if restrictii_msg:
        with open(output_file, 'w') as file:
            file.write(restrictii_msg)
    else:
        nivele_pare = noduri_pe_nivele_pare(n, parinti)
        scrie_noduri_pe_nivele_pare(output_file, nivele_pare)

</syntaxhighlight>