Pagini recente » Cod sursa (job #1876373) | Cod sursa (job #3342187) | Cod sursa (job #1873211) | Monitorul de evaluare | Cod sursa (job #3342505)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int NMAX = 1e5;
int n, ind;
int a[NMAX + 1];
int dp[NMAX + 1];
vector<int> g[NMAX + 1];
int st[NMAX + 1];
void DFS(int node, int dad = 0) {
if(a[node]) {
dp[node] = dp[st[ind - a[node] + 1]] + 1;
}
st[++ind] = node;
for(int next_node : g[node]) {
if(next_node != dad) {
DFS(next_node, node);
}
}
ind--;
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
fin >> n;
for(int i = 1; i <= n; i++) {
fin >> a[i];
}
for(int i = 1; i <= n - 1; i++) {
int a, b;
fin >> a >> b;
g[a].push_back(b);
g[b].push_back(a);
}
DFS(1);
for(int i = 1; i <= n; i++) {
fout << dp[i] << ' ';
}
return 0;
}