Pagini recente » Cod sursa (job #2343552) | Cod sursa (job #408531) | Cod sursa (job #83719) | Cod sursa (job #3224014) | Cod sursa (job #2647040)
#include <bits/stdc++.h>
using namespace std;
#define NMAX 100005
int n,k[NMAX],ans[NMAX];
vector<int> v[NMAX], st;
void dfs(int nod){
if (k[nod] == 0) ans[nod] = 0;
else ans[nod] = ans[st[st.size() - k[nod]]] + 1;
st.push_back(nod);
for (auto it : v[nod]){
dfs(it);
}
st.pop_back();
}
int main()
{
freopen("cerere.in","r",stdin);
freopen("cerere.out","w",stdout);
cin >> n;
for (int i=1;i<=n;i++) cin >> k[i];
for (int i=1;i<n;i++){
int a,b;
cin >> a >> b;
v[a].push_back(b);
}
dfs(1);
for (int i=1;i<=n;i++) cout << ans[i] << ' ';
cout << '\n';
return 0;
}