Pagini recente » Cod sursa (job #829776) | Cod sursa (job #2176786) | Cod sursa (job #22093) | Cod sursa (job #1944072) | Cod sursa (job #2358688)
#include<fstream>
#include<vector>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,x,y;
int stiva[100001],D[100001],v[100001],K[100001];
vector <int> L[100001];
void dfs(int nod,int pas){
stiva[pas]=nod;
D[nod]=1+D[stiva[pas-K[nod]]];
v[nod]=1;
for(int i=0;i<L[nod].size();i++){
if(v[L[nod][i]]==0){
v[L[nod][i]]=1;
dfs(L[nod][i],pas+1);
}
}
}
int main(){
fin>>n;
for(int i=1;i<=n;i++)
fin>>K[i];
for(int i=1;i<=n-1;i++){
fin>>x>>y;
L[x].push_back(y);
L[y].push_back(x);
}
for(int i=1;i<=n;i++)
if(K[i]==0){
dfs(i,0);
break;
}
for(int i=1;i<=n;i++)
fout<<D[i]-1<<" ";
return 0;
}