Pagini recente » Cod sursa (job #2638419) | Cod sursa (job #1450809) | Cod sursa (job #2792865) | Cod sursa (job #2991039) | Cod sursa (job #2212335)
#include <fstream>
#include <vector>
using namespace std;
vector <int> v[100002];
int n,k,a,b,I,tati[100002],val[100002],cere[100002],nivel[100002];
void dfs(int start, int loc)
{
int i;
nivel[loc]=start;
for(i=0;i<v[start].size();i++)
{
k=v[start][i];
if(cere[k])
val[k]=val[nivel[loc-cere[k]+1]]+1;
dfs(k,loc+1);
}
}
int main()
{
ifstream f("cerere.in");
ofstream g("cerere.out");
int i;
f>>n;
for(i=1;i<=n;i++)
f>>cere[i];
for(i=2;i<=n;i++)
{
f>>a>>b;
v[a].push_back(b);
tati[b]=a;
}
for(i=1;i<=n;i++)
if(tati[i]==0)
{
I=i;
break;
}
dfs(I,1);
for(i=1;i<=n;i++)
g<<val[i]<<' ';
return 0;
}