Pagini recente » Cod sursa (job #2964275) | Cod sursa (job #982792) | Cod sursa (job #1247691) | Cod sursa (job #2775974) | Cod sursa (job #2566741)
#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];
bool root[NMAX];
int top,lvl;
void dfs(int nod) {
if(K[nod])
ans[nod]=ans[stiva[top-K[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;
}