Pagini recente » Cod sursa (job #543790) | Cod sursa (job #1189010) | Cod sursa (job #2945169) | Cod sursa (job #486752) | Cod sursa (job #1781463)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
#define MAX 100000
vector <int> G[MAX];
int n,cer[MAX],tata[MAX],rez[MAX],viz[MAX],dim;
int stv[MAX];
void cereri(int nod)
{
dim++;
stv[dim]=nod;
rez[nod]=rez[stv[dim-cer[nod]]]+1;
for(auto it:G[nod])
{
cereri(it);
}
dim--;
}
int main()
{
int x,y;
f>>n;
for(int i=1;i<=n;i++)
f>>cer[i];
for(int i=1;i<n;i++)
{
f>>x>>y;
G[x].push_back(y);
tata[y]=x;
}
for(int i=1;i<=n;i++)
if(tata[i]==0)
{
cereri(i);
break;
}
for(int i=1;i<=n;i++)
g<<rez[i]<<" ";
return 0;
}