Pagini recente » Cod sursa (job #157398) | Cod sursa (job #3160841) | Cod sursa (job #1822761) | Cod sursa (job #1376717) | Cod sursa (job #1786839)
#include <iostream>
#include <fstream>
#include <stdlib.h>
#define nmax 100099
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int* g[nmax];
int k[nmax],rez[nmax],s[nmax],sf[nmax];
char viz[nmax];
int main()
{int i,n,a,b,rad,fiu;
fin>>n;
for(i=1;i<=n;i++)
{fin>>k[i];
g[i]=(int*)realloc(g[i],sizeof(int));g[i][0]=0;
}
for(i=1;i<=n-1;i++)
{fin>>a>>b;
g[a][0]++;
g[a]=(int*)realloc(g[a],(g[a][0]+1)*sizeof(int));
g[a][g[a][0]]=b;
viz[b]=1;
}
rad=1;while(viz[rad])rad++;
//dfs:
int vf=1;
s[1]=rad;vf=1;sf[1]=1;
while(vf)
{//for(i=1;i<=vf;i++)cout<<s[i]<<' ';cout<<endl;
rad=s[vf]; fiu=sf[vf];
if(k[rad])rez[rad]=rez[s[vf-k[rad]]]+1;
if(fiu<=g[rad][0])
{s[++vf]=g[rad][fiu];
sf[vf]=1;
sf[vf-1]=fiu+1;
}else vf--;
}
for(i=1;i<=n;i++)fout<<rez[i]<<' ';
}