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