Pagini recente » Cod sursa (job #3226098) | Cod sursa (job #1132490) | Cod sursa (job #2101533) | Cod sursa (job #2925746) | Cod sursa (job #2641737)
#include <cstdio>
#include <vector>
#include <cctype>
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);
}
char buffer[1 << 17];
int sz = (1 << 17) , crs = (1 << 17);
void get_int(int &n)
{
for( ; crs < sz && !isdigit(buffer[crs]) ; crs ++);
if(crs == sz)
{
fread(buffer , sz , 1 , stdin);
for(crs = 0 ; crs < sz && !isdigit(buffer[crs]) ; crs ++);
}
n = 0;
for( ; crs < sz && isdigit(buffer[crs]) ; crs ++)
n = n * 10 + buffer[crs] - '0';
if(crs == sz)
{
fread(buffer , sz , 1 , stdin);
for(crs = 0 ; crs < sz && isdigit(buffer[crs]) ; crs ++)
n = n * 10 + buffer[crs] - '0';
}
}
int main()
{
freopen("cerere.in" , "r" , stdin);
freopen("cerere.out" , "w" , stdout);
int n , i , x , y;
get_int(n);
for(i = 1 ; i <= n ; i ++)
get_int(s[i]);
for(i = 1 ; i < n ; i ++)
{
get_int(x);
get_int(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;
}