Pagini recente » Cod sursa (job #809693) | Cod sursa (job #152710) | Cod sursa (job #199479) | Cod sursa (job #2310026) | Cod sursa (job #2558350)
#include <iostream>
#include <cstdio>
#include <vector>
#include <fstream>
#define UNVISITED -1
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int N = 100000;
int pas[N+1], urm[N+1];
int d[N+1], in[N+1];
vector <int> g[N+1];
vector <int> stiva;
void dfs(int node){
stiva.push_back(node);
int poz = (int)stiva.size() - pas[node] - 1;
urm[node] = stiva[poz];
for(int p=0; p<(int)g[node].size(); p++){
int next = g[node][p];
dfs(next);
}
stiva.pop_back();
}
void solve(int node){
int next = urm[node];
if(node == next){
d[node] = 0;
return;
}
if(d[next] == UNVISITED)
dfs(next);
d[node] = d[next] + 1;
}
int main()
{
int n,i,x,y,root;
fin >> n;
for(i=1; i<=n; i++){
fin >> pas[i];
d[i] = UNVISITED;
}
for(i=1; i<n; i++){
fin >> x >> y;
in[y]++;
g[x].push_back(y);
}
root = 1;
while(root <= n && in[root] > 0)
root++;
dfs(root);
for(i=1; i<=n; i++)
if(d[i] == UNVISITED)
solve(i);
for(i=1; i<=n; i++)
fout << d[i] << " ";
return 0;
}