Pagini recente » Cod sursa (job #1217268) | Cod sursa (job #1542148) | Cod sursa (job #437235) | Cod sursa (job #2289116) | Cod sursa (job #2566738)
#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 100005
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
vector <int> g[NMAX];
int n,i,K[NMAX];
int stiva[NMAX],ans[NMAX],parent[NMAX];
bool root[NMAX];
int top,lvl;
void dfs(int nod) {
if(K[nod]) {
modg[stiva[top-K[nod]]].push_back(nod);
parent[nod]=stiva[top-K[nod]];
ans[nod]=ans[parent[nod]]+1;
}
for(int i=0; i<(int)g[nod].size(); i++) {
int aux=g[nod][i];
stiva[++top]=aux;
dfs(aux);
top--;
}
}
int main() {
fin>>n;
for(i=1; i<=n; i++) {
fin>>K[i];
root[i]=true;
}
for(i=1; i<n; i++) {
int x,y;
fin>>x>>y;
g[x].push_back(y);
root[y]=false;
}
for(i=1; i<=n; i++)
if(root[i]==true) {
stiva[++top]=i;
break;
}
dfs(i);
for(i=1; i<=n; i++)
fout<<ans[i]<<" ";
return 0;
}