Editing 2749 – Tată (section)

= Exemplul 2: =
Intrare
 5
 2 0 2 5 4
Ieșire
 NU

=== Explicație ===
Nodul <code>4</code> are ca tată pe <code>5</code>, iar nodul <code>5</code> are ca tată pe <code>4</code>

== Exemplul 3: ==
Intrare
 100001
Consola
 Datele nu corespund restrictiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python3">
def sunt_restrictii_incorecte(n, t):
    # Verifică restricțiile pentru n
    if not (1 <= n <= 100000):
        return True

    # Verifică restricțiile pentru t[i]
    for i in range(n):
        if not (0 <= t[i] <= n):
            return True

    return False

def este_vector_de_tati(t):
    n = len(t)

    # Verifică dacă restricțiile sunt respectate
    if sunt_restrictii_incorecte(n, t):
        return False

    # Verifică dacă rădăcina arborelui este corectă
    if t[0] != 0:
        return False

    # Verifică dacă fiecare element al vectorului este un index valid
    for i in range(1, n):
        if not (0 <= t[i] < n):
            return False

    return True

# Citeste numarul n
n = int(input("Introduceti numarul n: "))

# Ieși din program dacă restricțiile pentru n nu sunt respectate
if n < 1 or n > 100000:
    print("Datele nu corespund restrictiilor impuse")
else:
    # Citeste vectorul t
    t = list(map(int, input("Introduceti vectorul t: ").split()))

    # Verifica daca restrictiile sunt respectate si afiseaza mesajul corespunzator
    if sunt_restrictii_incorecte(n, t):
        print("Datele nu corespund restrictiilor impuse")
    else:
        # Verifica daca t este vector de tati si afiseaza rezultatul
        if este_vector_de_tati(t):
            print("DA")
        else:
            print("NU")

</syntaxhighlight>