Cerința

Se dă vectorul de tați al unui arbore cu rădăcină cu n noduri. Determinați câte perechi de noduri neterminale distincte p q din arbore au proprietatea că subarborele cu rădăcina în p și cel cu rădăcina în q au același număr de noduri.

Date de intrare

Fișierul de intrare subarbori.in conține pe prima linie numărul de noduri n. 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 subarbori.out va conține pe prima linie valoarea cerută.

Restricții și precizări

• 1 ≤ n ≤ 100
• în vectorul de tați rădăcina este marcată cu 0

Exemplu

subarbori.in

8
4 3 0 3 2 1 2 1

subarbori.out

1

Explicație

Singura pereche de noduri neterminale pentru care subarborii au același număr de noduri este 2 1. Subarborii cu rădăcina în 2 și 1 au câte 3 noduri fiecare.

#include <bits/stdc++.h>
using namespace std;

ifstream cin("subarbori.in");
ofstream cout("subarbori.out");

int n , k , t[101] , v[101] , cnt , f[101];

int main()
{
    cin >> n;
    for(int i = 1 ; i <= n ; i++)
        cin >> t[i];

    for(int i = 1 ; i <= n ; i++)
    {
        int p = t[i];
        while(p)
        {
            f[p]++;
            p = t[p];
        }
    }
    for(int i = 1 ; i <= n ; i++)
        for(int j = i + 1 ; j <= n ; j++)
            if(f[i] == f[j] && f[i] != 0) cnt++;
    cout << cnt;

}