Pagini recente » Cod sursa (job #1404606) | Cod sursa (job #1024480) | Cod sursa (job #2229811) | Cod sursa (job #2612597) | Cod sursa (job #1068017)
#include <iostream>
#include <fstream>
#include <vector>
#define N_MAX 100001
using namespace std;
vector<int> v[N_MAX];
vector<int> stiva;
int c[N_MAX],k[N_MAX],t[N_MAX];
int dfs(int x){
stiva.push_back(x);
if(k[x]!=0)
c[x]=c[stiva[stiva.size()-k[x]-1]]+1;
for(int i=0;i<v[x].size();i++){
//cout<<x<<"are"<<v[x][i];
dfs(v[x][i]);
}
stiva.pop_back();
}
int main()
{
int n,i,a,b;
ifstream f("cerere.in");
ofstream g("cerere.out");
f>>n;
for(i=1;i<=n;i++){
f>>k[i];
}
for(i=1;i<=n-1;i++){
f>>a>>b;
t[b]=a;
v[a].push_back(b);
}
for(i=1;i<=n;i++)
if(t[i]==0)
break;
dfs(i);
for(i=1;i<=n;i++)
g<<c[i]<<" ";
return 0;
}