Pagini recente » Cod sursa (job #2295856) | Cod sursa (job #2275656) | Cod sursa (job #2849549) | Cod sursa (job #1726231) | Cod sursa (job #992249)
Cod sursa(job #992249)
#include <fstream>
#include <vector>
#include <string.h>
using namespace std;
ifstream f("cerere.in");
ofstream g("cerere.out");
int Ancestor[100002];
int N;
int Sol[100002];
int Nodes[100002],ind;
bool Use[100002];
int TT[100002];
vector <int> G[100002];
void Read()
{
int i;
f>>N;
for(i=1;i<=N;i++)
f>>Ancestor[i];
for(i=1;i<=N-1;i++)
{
int x,y;
f>>x>>y;
TT[y]=x;
G[x].push_back(y);
}
}
void DFS1(int nod)
{
unsigned int i;
Use[nod]=1;
Nodes[++ind]=nod;
if(Ancestor[nod]!=0)
Sol[nod]=Sol[Nodes[ind-Ancestor[nod]]]+1;
for(i=0;i<=G[nod].size();i++)
{
int Vecin=G[nod][i];
if(!Use[Vecin])
DFS1(Vecin);
}
ind--;
}
int main()
{
Read();
for(int i=1;i<=N;i++)
if(TT[i]==0)
{
DFS1(i);
break;
}
for(int i=1;i<=N;i++)
g<<Sol[i]<<" ";
g<<"\n";
return 0;
}