Pagini recente » Cod sursa (job #1699353) | Cod sursa (job #689289) | Cod sursa (job #2444149) | Cod sursa (job #3313813) | Cod sursa (job #3313306)
#include <bits/stdc++.h>
using namespace std;
int dp[200005];
int st[200005];
vector<int>adj[200005];
int k[200005];
int f[200005];
void dfs(int node, int l)
{
st[l]=node;
if(k[node]!=0)
dp[node]=dp[st[l-k[node]]]+1;
else
dp[node]=0;
for(auto i : adj[node])
{
dfs(i, l+1);
}
}
int main()
{
ifstream cin ("cerere.in");
ofstream cout ("cerere.out");
int n;
cin >> n;
for(int i=1; i<=n; i++)
cin >> k[i];
for(int i=1; i<=n-1; i++)
{
int a, b;
cin >> a >> b;
adj[a].push_back(b);
f[b]++;
}
int poz;
for(int i=1; i<=n; i++)
{
if(f[i]==0)
poz=i;
}
dfs(poz, 1);
for(int i=1; i<=n; i++)
cout << dp[i] << ' ';
return 0;
}