Pagini recente » Cod sursa (job #2037999) | Cod sursa (job #2899379) | Cod sursa (job #1718370) | Cod sursa (job #44098) | Cod sursa (job #35779)
Cod sursa(job #35779)
#include <stdio.h>
#include <vector>
using namespace std;
int n, d[100001], sol[100001], used[100001], back[100001], cnt;
vector<int> v[100001];
void dfs(int nod);
int main()
{
freopen("cerere.in","r",stdin);
freopen("cerere.out","w",stdout);
int i, a, b;
scanf("%d", &n);
for(i = 1; i <= n; ++i)
{
scanf("%d", &d[i]);
}
for(i = 1; i < n; ++i)
{
scanf("%d%d", &a, &b);
v[a].push_back(b);
}
dfs(1);
for(i = 1; i < n; ++i)
printf("%d ", sol[i]);
printf("%d\n", sol[n]);
return 0;
}
void dfs(int nod)
{
int i;
used[nod] = 1;
back[++cnt] = nod;
if(!d[nod])
{
sol[nod] = 0;
}
else
{
sol[nod] = sol[back[cnt - d[nod]]] + 1;
}
for(i = 0; i < v[nod].size(); ++i)
{
if(!used[v[nod][i]])
{
dfs(v[nod][i]);
}
}
--cnt;
}