Pagini recente » Cod sursa (job #1735591) | Cod sursa (job #3276770) | Cod sursa (job #1518405) | Cod sursa (job #1137235) | Cod sursa (job #2011494)
#include <iostream>
#include <fstream>
#include <vector>
#define N 100010
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
vector<int> vec[N];
int v[N],st[N],sol[N],arb[N],i,t,j,n,k,m;
void dfs(int x)
{
st[++t]=x;
if(v[x])
sol[x]=sol[st[t-v[x]]]+1;
for(int ind=0;ind<vec[x].size();ind++)
dfs(vec[x][ind]);
--t;
}
int main()
{
f>>n;
for(i=1;i<=n;i++)
f>>v[i];
for(i=1;i<n;i++)
{
f>>m>>k;
vec[m].push_back(k);
arb[k]=1;
}
for(i=1;arb[i];i++);
dfs(i);
for(i=1;i<=n;i++)
g<<sol[i]<<' ';
}