Pagini recente » Cod sursa (job #2136725) | Cod sursa (job #2226617) | Cod sursa (job #2475095) | Cod sursa (job #3148784) | Cod sursa (job #2599664)
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 100005;
int cer[MAXN], ans[MAXN];
bool fiu[MAXN];
vector<int> graf[MAXN], lant;
void dfs(int nod){
lant.push_back(nod);
if((int)graf[nod].size() == 0) return;
for(auto x: graf[nod]){
if(cer[x] > 0){
int stramos = lant[lant.size() - cer[x]];
ans[x] = ans[stramos] + 1;
}
dfs(x);
lant.pop_back();
}
}
int main()
{
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n;
fin >> n;
for(int i = 1; i <= n; ++i)
fin >> cer[i];
for(int i = 1; i < n; ++i){
int x, y;
fin >> x >> y;
graf[x].push_back(y);
fiu[y] = 1;
}
int root = 0;
for(int i = 1; i <= n; ++i){
if(!fiu[i]){
root = i;
break;
}
}
dfs(root);
for(int i = 1; i <= n; ++i) fout << ans[i] << " ";
return 0;
}