Pagini recente » Cod sursa (job #1298842) | Cod sursa (job #2905232) | Cod sursa (job #2368694) | Cod sursa (job #2074383) | Cod sursa (job #1306831)
#include <fstream>
#include <vector>
#define DIM 100002
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
vector <int> v[DIM];
int n,s[DIM],t[DIM],sol[DIM],nv[DIM],viz[DIM],niv;
void DFS(int x){
nv[++niv]=x;
if(s[x]!=0){
sol[x]=sol[ nv[ niv - s[x] ] ]+1;
}
for(int i=0;i<v[x].size();i++)
DFS(v[x][i]);
niv--;
}
int main(){
fin>>n;
for(int i=1;i<=n;i++){
fin>>s[i];
}
for(int i=2;i<=n;i++){
int x,y;
fin>>x>>y;
v[x].push_back(y);
t[y]=x;
}
for(int i=1;i<=n;i++)
if(!t[i])
DFS(i);
for(int i=1;i<=n;i++)
fout<<sol[i]<<" ";
fin.close();fout.close();
return 0;
}