Pagini recente » Cod sursa (job #60354) | Cod sursa (job #1904910) | Cod sursa (job #1557577) | Cod sursa (job #1589700) | Cod sursa (job #1054187)
#include <cstdio>
#include <vector>
using namespace std;
FILE*f=fopen("cerere.in","r");
FILE*h=fopen("cerere.out","w");
int s[100001],t[100001],n,e[100001],k,r[100001];
vector<int> v[100001];
inline void dfs( int x ){
s[++k]=x;
if ( t[x] )
r[x]=r[s[k-t[x]]]+1;
for ( int i=0;i<v[x].size();++i )
dfs(v[x][i]);
--k;
}
int main()
{
fscanf(f,"%d",&n);
for ( int i=1;i<=n;++i )
fscanf(f,"%d",&t[i]);
for ( int i=1;i<n;++i ){
int a,b;
fscanf(f,"%d%d",&a,&b);
v[a].push_back(b);
e[b]=1;
}
for ( int i=1;i<=n;++i )
if ( e[i]==0 ){dfs(i);break;}
for ( int i=1;i<=n;++i )
fprintf(h,"%d ",r[i]);
return 0;
}