Pagini recente » Cod sursa (job #551028) | Cod sursa (job #2093755) | Cod sursa (job #1097185) | Cod sursa (job #859261) | Cod sursa (job #1784516)
#include <iostream>
#include <fstream>
using namespace std;
#define MAXN 100010
int v[MAXN],tata[MAXN],stramos[MAXN],n,x,y,viz[MAXN];
ifstream f("cerere.in");
ofstream g("cerere.out");
void df(int nod)
{
int stra=nod;
viz[nod]=1;
for(int i=1;i<=stramos[nod];i++)
{
stra=tata[stra];
}
if(!viz[stra])
df(stra);
if(stramos[nod]==0)
stra=0;
v[nod]=v[stra]+1;
}
int main()
{
f>>n;
for(int i=1;i<=n;i++)
f>>stramos[i];
for(int i=1;i<n;i++)
{
f>>x>>y;
tata[y]=x;
}
v[0]=-1;
for(int i=1;i<=n;i++)
{
if(!viz[i])
df(i);
}
for(int i=1;i<=n;i++)
g<<v[i]<<' ';
return 0;
}