Pagini recente » Cod sursa (job #1705235) | Cod sursa (job #2178857) | Cod sursa (job #2712377) | Cod sursa (job #2541678) | Cod sursa (job #3341347)
#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, sizeof(ans));
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;
}