Cerința

Se dă vectorul de tați al unui arbore cu rădăcină cu n noduri și un nod k. Fiecare nod al arborelui are asociată o valoare numerică întreagă. Determinați suma valorilor asociate nodurilor din subarborele cu rădăcina în k.

Date de intrare

Fișierul de intrare subarbore2.in conține pe prima linie numărul de noduri n și nodul k. Pe a doua linie se află vectorul de tați al arborelui, valorile fiind separate prin spații. Pe linia a treia se află, în ordine, valorile asociate nodurilor din arbore.

Date de ieșire

Fișierul de ieșire subarbore2.out va conține pe prima linie valoarea cerută.

Restricții și precizări

1 ≤ n ≤ 100
1 ≤ k ≤ n
în vectorul de tați rădăcina este marcată cu 0
valorile asociate nodurilor din arbore sunt numere întregi din intervalul [-1000,1000]

Exemplu

subarbore2.in

8 4
4 3 0 3 2 1 4 1
-3 -2 4 4 0 -3 3 2

subarbore2.out

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 Subarbore2:

#include <iostream>
#include <fstream>
using namespace std;

ifstream fin("subarbore2.in");
ofstream fout("subarbore2.out");

int n , k , t[105], v[105] , sum[105];

void  dfs(int k)
{
    int s = v[k];
    for(int i = 1 ; i <= n ; ++i)
        if(t[i] == k)
        {
            dfs(i);
            s += sum[i];
        }
    sum[k] = s;
}

int main()
{
    fin >> n >> k;
    for(int i = 1 ; i <= n ; i ++)
        fin >> t[i];
    for(int i = 1 ; i <= n ; i ++)
        fin >> v[i];
    dfs(k);
    fout << sum[k];
    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 #647 Subarbore2

Pe această pagină găsești rezolvarea de 100 de puncte pentru problema #647 Subarbore2 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!

Rezolvare completă PbInfo #647 Subarbore2