Pagini recente » Cod sursa (job #2535013) | Cod sursa (job #2670377) | Cod sursa (job #2768362) | Cod sursa (job #1131042) | Cod sursa (job #2541224)
#include <bits/stdc++.h>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int n;
int k[100001];
int parent[100001];
int who[100001];
int cnt[100001];
//vector < int > v[100001];
int main()
{
f>>n;
int i;
for(i=1;i<=n;i++)
f>>k[i];
for(i=1;i<=n-1;i++)
{
int n1,n2;
f>>n1>>n2;
// v[n1].push
parent[n2]=n1;
}
for(i=1;i<=n;i++)
{
int nod=i;
for(int j=1;j<=k[i];j++)
nod=parent[nod];
who[i]=nod;
}
// for(i=1;i<=n;i++)
// cnt[i]=-1;
for(i=1;i<=n;i++)
{
int cont=0;
int nod=i;
while(k[nod]!=0)
{
nod=who[nod],++cont;
if(cnt[nod]!=0)
{
cont+=cnt[nod];
break;
}
}
cnt[i]=cont;
g<<cont<<' ';
}
return 0;
}