Editing 0652 – Afișare Fii

= Cerința =
Se dă vectorul de tați al unui arbore cu rădăcină cu <code>n</code> noduri și <code>k</code> noduri distincte din arbore. Afișați fiii fiecăruia dintre cele <code>k</code> noduri.

= Date de intrare =
Fișierul de intrare <code>afisarefiiIN.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. Linia a trei conține numărul <code>k</code>, iar linia a patra <code>k</code> noduri, <code>x[1]</code>, <code>x[2]</code>, … , <code>x[k]</code>.

= Date de ieșire =
Fișierul de ieșire <code>afisarefiiOUT.txt</code> va conține <code>k</code> linii. Linia <code>i</code> va conține numărul de fii ai lui <code>x[i]</code> urmat de un spațiu și de lista fiilor lui <code>x[i]</code>, separați și ei 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 ≤ k ≤ n ≤ 100</code>
* în vectorul de tați rădăcina este marcată cu <code>0</code>
* dacă un nod <code>x[i]</code> nu are fii, linia corespunzătoare din fișierul de ieșire va contine doar valoarea <code>0</code>

= Exemplul 1: =
<code>afisarefiiIN.txt</code>
 8
 4 3 0 3 2 1 2 1
 4
 3 4 7 1
<code>afisarefiiOUT.txt</code>
 2 2 4 
 1 1 
 0 
 2 6 8 

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

== Rezolvare ==
<syntaxhighlight lang="python3">
def verifica_restrictii(n, k):
    if not (1 <= k <= n <= 100):
        return "Datele nu corespund restrictiilor impuse"
    return None

def afisare_fii(n, tati, k, noduri_selectate):
    restrictii_mesaj = verifica_restrictii(n, k)
    if restrictii_mesaj:
        return restrictii_mesaj

    rezultat = []

    for nod in noduri_selectate:
        if not (1 <= nod <= n):
            return "Datele nu corespund restrictiilor impuse"

        fii = [i for i, parinte in enumerate(tati, 1) if parinte == nod]
        rezultat.append((len(fii), fii))

    return rezultat

# Citirea datelor de intrare
with open("afisarefiiIN.txt", "r") as f:
    n = int(f.readline().strip())
    tati = list(map(int, f.readline().strip().split()))
    k = int(f.readline().strip())
    noduri_selectate = list(map(int, f.readline().strip().split()))

# Verificarea restricțiilor și obținerea rezultatului
restrictii_mesaj = verifica_restrictii(n, k)
if restrictii_mesaj:
    with open("afisarefiiOUT.txt", "w") as f:
        f.write(restrictii_mesaj)
else:
    rezultat = afisare_fii(n, tati, k, noduri_selectate)
    with open("afisarefiiOUT.txt", "w") as f:
        if isinstance(rezultat, str):
            f.write(rezultat)
        else:
            for nr_fii, fii in rezultat:
                f.write(f"{nr_fii} {' '.join(map(str, fii))}\n")

</syntaxhighlight>