Pagini recente » Cod sursa (job #484226) | Cod sursa (job #1028719) | Cod sursa (job #1563466) | Cod sursa (job #1920075) | Cod sursa (job #2116090)
#include<cstdio>
#include<vector>
#include<cmath>
#include<queue>
#define DN 100005
#define pb push_back
using namespace std;
int n,p,q,dp[DN],k[DN],d,f,z[DN],niv[DN],l,pr[DN];
vector<int>v[DN];
void dfs(int nod)
{
z[niv[nod]]=nod;
if(k[nod]!=0)
{
f=k[nod];
l=niv[nod]-k[nod];
d=z[l];
dp[nod]=dp[d]+1;
}
for(auto i:v[nod])
{
niv[i]=niv[nod]+1;
dfs(i);
}
}
int main()
{
freopen("cerere.in","r",stdin);
freopen("cerere.out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&k[i]);
for(int i=1;i<n;i++)
{
scanf("%d%d",&p,&q);
v[p].pb(q);
pr[q]=p;
}
for(int i=1;i<=n;i++)
if(pr[i]==0)
{
niv[i]=1;
dfs(i);
break;
}
for(int i=1;i<=n;i++)
printf("%d ",dp[i]);
}