Pagini recente » 1235123 | Cod sursa (job #2013168) | Monitorul de evaluare | Profil Marosaru | Cod sursa (job #1736479)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int MAXN = 1e5 + 5;
vector <int> graf[MAXN];
int stiva[MAXN], rez[MAXN];
int n, v[MAXN], root, dim;
void dfs(int nod)
{
stiva[++dim] = nod;
if( v[nod] )
rez[nod] = rez[stiva[dim - v[nod]]] + 1;
for(auto it: graf[nod])
dfs(it);
--dim;
}
int main()
{
fin>>n;
for(int i=1; i<=n; i++) fin>>v[i];
root = n * (n+1) / 2;
for(int i=1; i<n ; i++)
{
int x,y;
fin>>x>>y;
graf[x].push_back(y);
root -= y;
}
dfs(root);
for(int i=1; i<=n; i++) fout<<rez[i]<<" ";
}