# PBinfo #2749 – tata – Rezolvare C++

Sursa: Codul lui Andrei
Limbaj: C++
Autor rezolvare: Darius

## Enunț

Cerința
Se dă un vector t=(t[1], t[2], ..., t[n]) care memorează numere naturale cuprinse între 0 și n. Să se verifice dacă t este sau nu vector de tați asociat unui arbore cu rădăcină.

Date de intrare
Programul citește de la tastatură numărul n, iar apoi n numere naturale, separate prin spații.

Date de ieșire
Programul va afișa pe ecran mesajul DA, dacă t este vector de tați, sau mesajul NU în caz contrar.

Restricții și precizări

1 ≤ n ≤ 100.000
0 ≤ t[i] ≤ n

Exemplul 1:
Intrare

5
0 1 1 3 3
Ieșire

DA
Explicație
Vectorul corespunde arborelui cu rădăcină din imagine.



Exemplul 2:
Intrare

5
2 0 2 5 4
Ieșire

NU
Explicație
Nodul 4 are ca tată pe 5, iar nodul 5 are ca tată pe 4

## Cod sursă C++

```cpp
#include <bits/stdc++.h>
#define N 100005
using namespace std;
int n, rad;
int viz[N];
vector<int> h[N];
void DFS(int k)
{
    viz[k] = 1;
    for (auto i : h[k])
        if (viz[i] == 0)
            DFS(i);
}
int IsTree()
{
    DFS(rad);
    for (int i = 1; i <= n; i++)
        if (viz[i] == 0)
            return 0;
    return 1;
}
int main()
{
    int i, x, nrZ;
    cin >> n;
    nrZ = 0;
    for (i = 1; i <= n; i++)
    {
        cin >> x;
        if (x == 0)
        {
            nrZ++;
            rad = i;
        }
        else h[x].push_back(i);
    }
    if (nrZ != 1)
    {
        cout << "NU\n";
        return 0;
    }
    if (IsTree()) cout << "DA\n";
    else cout << "NU\n";
    return 0;
}
```

## Resurse

- [Pagina originală](https://codulluiandrei.ro/rezolvari/pbinfo/2749)
- [Sursa C++](https://codulluiandrei.ro/rezolvari/pbinfo/2749/cod.cpp)
- [Sursa pe GitHub](https://raw.githubusercontent.com/codulluiandrei/pbinfo/main/pbinfo-2749/main.cpp)
- [Cursuri de informatică pentru BAC](https://codulluiandrei.ro/academia/info-curs/pachet-cpp/)