Pagini recente » Cod sursa (job #809845) | Cod sursa (job #1094181) | Cod sursa (job #99976) | Cod sursa (job #888247) | Cod sursa (job #1867869)
#include<fstream>
using namespace std;
int n,nr,pp,i,j,x,y,k,vf[200005],lst[100005],pred[100005],urm[100005],v[100005],sol[100005];
bool viz[100005],c[100005];
void adauga(int x, int y)
{
nr++;
vf[nr]=y;
urm[nr]=lst[x];
lst[x]=nr;
}
void dfs(int x)
{
viz[x]=true;
pred[pp]=x;
int poz=lst[x],y,i,k;
if(v[x]==0)
sol[x]=0;
else
{
k=x;
k=pred[pp-v[x]];
sol[x]=sol[k]+1;
}
while(poz!=0)
{
y=vf[poz];
if(!viz[y])
{
pp++;
dfs(y);
pp--;
}
poz=urm[poz];
}
}
int main()
{
ifstream f("cerere.in");
ofstream g("cerere.out");
f>>n;
for(i=1; i<=n; i++)
f>>v[i];
for(i=1; i<=n-1; i++)
{
f>>x>>y;
adauga(x,y);
c[y]=true;
}
for(i=1; i<=n; i++)
if(c[i]==false)
{
pp=1;
dfs(i);
break;
}
for(i=1; i<=n; i++)
g<<sol[i]<<" ";
return 0;
}