Pagini recente » Cod sursa (job #2973349) | Cod sursa (job #1181004) | Cod sursa (job #1029361) | Cod sursa (job #1703615) | Cod sursa (job #1054307)
#include<fstream>
#include<vector>
#define dim 100007
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
vector<int>G[dim];
int viz[dim];
int niv[dim],K[dim],x,y,i,n,sol[dim];
void dfs(int nod ,int nivel){
niv[nivel]=nod;
if(K[nod])
sol[nod]=sol[niv[nivel-K[nod]]]+1;
for(int i=0;i<G[nod].size();++i) {
dfs(G[nod][i],nivel+1);
}
}
int main() {
f>>n;
for(i=1;i<=n;++i){
f>>x;
K[i]=x;
}
for(i=1;i<n;++i){
f>>x>>y;
G[x].push_back(y);
viz[y]=1;
}
for(i=1;i<=n;i++){
if(!viz[i])
dfs(i,1);
}
for(i=1;i<=n;++i){
g<<sol[i]<<" ";
}
return 0;
}