Pagini recente » Cod sursa (job #2266300) | Cod sursa (job #1810662) | Cod sursa (job #217122) | Cod sursa (job #3235517) | Cod sursa (job #2641735)
#include <cstdio>
#include <vector>
using namespace std;
const int NMAX = 100000;
vector <int> G[NMAX + 5] , T[NMAX + 5];
int st[NMAX + 5] , s[NMAX + 5] , ans[NMAX + 5] , f[NMAX + 5] , root;
void dfs1(int u , int lvl)
{
st[lvl] = u;
if(s[u] != 0)
T[st[lvl - s[u]]].push_back(u);
else
T[0].push_back(u);
for(int j = 0 ; j < G[u].size() ; j ++)
dfs1(G[u][j] , lvl + 1);
}
void dfs2(int u , int lvl)
{
ans[u] = lvl - 1;
for(int j = 0 ; j < T[u].size() ; j ++)
dfs2(T[u][j] , lvl + 1);
}
int main()
{
freopen("cerere.in" , "r" , stdin);
freopen("cerere.out" , "w" , stdout);
int n , i , x , y;
scanf("%d" , &n);
for(i = 1 ; i <= n ; i ++)
scanf("%d" , &s[i]);
for(i = 1 ; i < n ; i ++)
{
scanf("%d%d" , &x , &y);
G[x].push_back(y);
f[y] = 1;
}
for(i = 1 ; i <= n && !root ; i ++)
if(!f[i])
root = i;
dfs1(root , 0);
dfs2(0 , 0);
for(i = 1 ; i <= n ; i ++)
printf("%d " , ans[i]);
return 0;
}