Pagini recente » Cod sursa (job #846106) | Cod sursa (job #2769023) | Cod sursa (job #3170655) | Cod sursa (job #2537375) | Cod sursa (job #2494191)
#include <bits/stdc++.h>
#define Nmax 100005
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int N, nr;
int pos[Nmax], ans[Nmax], st[Nmax];
vector<int> G[Nmax];
void DFS(int node) {
++nr, st[nr] = node;
if (pos[node]) ans[node] = 1 + ans[st[nr - pos[node]]];
for (auto it: G[node])
DFS(it);
--nr;
}
int main()
{
f >> N;
for (int i = 1; i <= N; ++i)
f >> pos[i];
for (int i = 1; i < N; ++i) {
int x, y;
f >> x >> y;
G[x].push_back(y);
}
DFS(1);
for (int i = 1; i <= N; ++i)
g << ans[i] << " ";
return 0;
}