Cod sursa(job #3341346)

Utilizator Dascalu_LucaDascalu Luca Petru Dascalu_Luca Data 19 februarie 2026 08:14:25
Problema Cerere Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.17 kb
#include <bits/stdc++.h>
using namespace std;

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

int n,K[100005],cnt_membrii=0,cnt_pasi=0,k;
int tata[100005],ans[100005];

void citire(){
    fin>>n;
    for(int i=1;i<=n;i++){
        fin>>K[i];
    }
    int i,j;
    while(fin>>i>>j){
        tata[j]=i;//ne intereseaza directia de la fiu catre parinte(j->i), adica relatia exterioara
    }
}

void parcurgere(int start,int k){
    if(cnt_pasi==k){
        cnt_pasi=0;
        cnt_membrii++;
        if(ans[start]!=-1){
            cnt_membrii+=ans[start];
            return;
        }
        if(K[start]==0) return;
        parcurgere(start, K[start]);
    }
    else{
        cnt_pasi++;
        parcurgere(tata[start],k);
    }
    //fiecare fiu are un singur stramosi

}

int main()
{
    ios_base::sync_with_stdio(false);
    fin.tie(NULL);
    memset(ans, -1, 100005);
    citire();
    for(int i=1;i<=n;i++){
        cnt_membrii=0, cnt_pasi=0,k=K[i];
        if(k!=0){
            parcurgere(i,k);
            ans[i]=cnt_membrii;
            fout<<cnt_membrii<<" ";
        }
        else fout<<0<<" ";
    }
    return 0;
}