Editing 0640 – NrFii

= Cerința =
Se dă vectorul de tați al unui arbore cu rădăcină cu <code>n</code> noduri. Determinați nodul din arbore cu număr maxim de fii. Dacă în arbore sunt mai multe noduri cu număr maxim de fii, afișați-le pe toate, în ordine crescătoare.

= Date de intrare =
Fișierul de intrare <code>nrfiiIN.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>nrfiiOUT.txt</code> va conține pe prima linie nodurile din arbore cu număr maxim de fii, în ordine crescătoare, separate printr-un spațiu.Î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>nrfiiIN.txt</code>
 7
 4 1 7 0 7 7 1 
<code>nrfiiOUT.txt</code>
 7

= Explicație =
Nodul <code>7</code> are număr maxim de fii, <code>3</code>.

== Exemplul 2: ==
<code>nrfiiIN.txt</code>
 101
 4 1 7 0 7 7 1 
<code>nrfiiOUT.txt</code>
 Datele nu corespund restrictiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def citeste_arbore(file_path):
    with open(file_path, 'r') as file:
        n = int(file.readline().strip())
        tati = list(map(int, file.readline().split()))
    return n, tati

def verificare_restrictii(n):
    if 1 <= n <= 100:
        return True
    else:
        with open("nrfiiOUT.txt", "w") as file_out:
            file_out.write("Datele nu corespund restrictiilor impuse")
        return False

def determina_noduri_max_fii(n, tati):
    fii = [0] * (n + 1)
    for tata in tati:
        if tata != -1:
            fii[tata] += 1

    max_fii = max(fii)
    noduri_max_fii = [i for i in range(1, n + 1) if fii[i] == max_fii]

    return noduri_max_fii

def scrie_rezultat(file_path, noduri_max_fii):
    with open(file_path, 'w') as file:
        if noduri_max_fii:
            file.write(' '.join(map(str, noduri_max_fii)))
        else:
            file.write("Datele nu corespund restrictiilor impuse")

if __name__ == "__main__":
    file_input = "nrfiiIN.txt"
    file_output = "nrfiiOUT.txt"

    n, tati = citeste_arbore(file_input)

    if verificare_restrictii(n):
        noduri_max_fii = determina_noduri_max_fii(n, tati)
        scrie_rezultat(file_output, noduri_max_fii)

</syntaxhighlight>