Pagini recente » Cod sursa (job #1880532) | Cod sursa (job #1552277) | Cod sursa (job #515433) | Cod sursa (job #1879249) | Cod sursa (job #1700847)
# include <fstream>
# include <vector>
# define DIM 100010
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
vector <int> Lista[DIM];
int Marcat[DIM],v[DIM],d[DIM],sol[DIM],h[DIM],n,i,x,y,ns;
void dfs(int nc,int val){
int nv;
Marcat[nc]=1;
if(v[nc]==0)
d[val]=0;
else
d[val]=d[val-v[nc]]+1;
sol[nc]=d[val];
for(int i=0;i<Lista[nc].size();i++){
nv=Lista[nc][i];
if(Marcat[nv]==0)
dfs(nv,val+1);
}
}
int main () {
fin>>n;
for(i=1;i<=n;i++)
fin>>v[i];
for(i=1;i<=n-1;i++){
fin>>x>>y;
Lista[x].push_back(y);
h[y]=1;
}
for(i=1;i<=n;i++){
if(h[i]==0)
ns=i;
}
dfs(ns,1);
for(i=1;i<=n;i++)
fout<<sol[i]<<" ";
fout<<"\n";
return 0;
}