Pagini recente » Cod sursa (job #2896057) | Cod sursa (job #2624892) | Cod sursa (job #691730) | Cod sursa (job #2936883) | Cod sursa (job #965134)
Cod sursa(job #965134)
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>
#define N 100005
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n, m, v, k[N], s[N], poz, sol[N];
vector <int> a[N];
vector <bool> viz(N), t(N);
void dfs(int x)
{
s[++poz] = x;
viz[x] = 1;
if(k[x])
sol[x] = sol[s[poz - k[x]]] + 1;
for(unsigned i=0; i<a[x].size(); i++)
if(!viz[a[x][i]])
dfs(a[x][i]);
poz--;
}
int main()
{
fin>>n;
int x, y;
for(int i=1; i<=n; i++)
fin>>k[i];
for(int i=1; i<n; i++)
{
fin>>x>>y;
a[x].push_back(y);
t[y] = 1;
}
for(int i=1; i<=n; i++)
if(!t[i]) {v = i; break;}
dfs(v);
for(int i=1; i<=n; i++)
fout<<sol[i]<<' ';
return 0;
}