Rezolvare completă PbInfo #649 Subarbori

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.

Cum e corect?

cout < "As la info"; cout << "As la info"; cout >> "As la info";

Felicitări! Poți mai mult?

Avem sute de probleme pentru tine, fiecare cu explicații ușor de înțeles.

Greșit, dar nu-i bai!

Antrenează-te cu sutele de probleme pe care ți le-am pregătit. Îți explicăm fiecare problemă în parte.

Rezolvare

Iată rezolvarea de 100 de puncte pentru problema Subarbori:

#include <iostream>
#include <fstream>
using namespace std;
 
ifstream fin("subarbori.in");
ofstream fout("subarbori.out");
 
int n , t[105], cnt[105] ,term[105];
 
void dfs(int k)
{
    int s = 0;
    for(int i = 1 ; i <= n ; ++i)
        if(t[i] == k)
        {
            dfs(i);
            s += cnt[i];
        }
    cnt[k] = s + 1;
}
 
int main()
{
    fin >> n;
    for(int i = 1 ; i <= n ; i ++)
        fin >> t[i];
        
    int r = 0;
    for(int i = 1 ; i <= n ; i ++)
        term[i] = 1;
    for(int i = 1 ; i <= n ; i ++)
        if(t[i] == 0)
            r = i;
        else
            term[t[i]] = 0;
    
    dfs(r);
    int nrp = 0;
    for(int i = 1 ; i < n ; ++i)
        for(int j = i + 1 ; j <= n ; ++j)
            if(!term[i] && !term[j] && cnt[i] == cnt[j])
                nrp ++;
    fout << nrp;
    return 0;
}

Atenție

Enunțurile afișate pe această pagină aparțin exclusiv site-ului PbInfo. Astfel, pentru ștergerea conținutului, puteți să ne contactați la adresa Adresa de email.

Rezolvarea problemei #649 Subarbori

Pe această pagină găsești rezolvarea de 100 de puncte pentru problema #649 Subarbori de pe PbInfo.ro. Atenție: nu încurajăm copiatul codului! Totuși, credem cu tărie că analizarea unei soluții corecte este o metodă foarte ușoară de a învăța informatică, astfel că oferim sursele pentru peste 1500 de probleme de pe platforma PbInfo.ro.

Pentru rezolvări PbInfo de la peste 1500 de probleme, vă invităm să intrați pe site-ul nostru!