Pagini recente » Cod sursa (job #594597) | Cod sursa (job #1920166) | Cod sursa (job #1529533) | Cod sursa (job #3146743) | Cod sursa (job #774049)
Cod sursa(job #774049)
#include<fstream>
#include<vector>
#include<cstring>
using namespace std;
int i,j,n,m=0,x,y,g[100001],s[100001],ord[100001],nr,niv[100001],rad,tata[100001];
vector<int> a[100001];
int viz[100001];
void df(int x)
{
int i;
ord[++m]=x;
if(!s[x])
g[x]=0;
else
g[x]=g[ord[m-s[x]]]+1;
for(i=0;i<a[x].size();++i)
df(a[x][i]);
--m;
}
int main()
{
freopen("cerere.in","r",stdin);
freopen("cerere.out","w",stdout);
scanf("%d",&n);
for(i=1;i<=n;++i)
scanf("%d",&s[i]);
for(i=1;i<n;++i)
{
scanf("%d%d",&x,&y);
a[x].push_back(y);
++tata[y];
}
for(i=1;i<=n;++i)
if(!tata[i])
rad=i,i=n+1;
df(rad);
for(i=1;i<=n;++i)
printf("%d ",g[i]);
return 0;
}