Pagini recente » Cod sursa (job #1002708) | Cod sursa (job #870663) | Cod sursa (job #1091827) | Cod sursa (job #2779506) | Cod sursa (job #927050)
Cod sursa(job #927050)
#include<fstream>
#include<vector>
#define NMAX 100005
using namespace std;
int n,v[NMAX],TT[NMAX],A[NMAX],C[NMAX],L;
vector<int> G[NMAX];
void read()
{
ifstream fin("cerere.in");
fin>>n;
int i,x,y;
for(i=1;i<=n;i++)
fin>>v[i];
for(i=1;i<n;i++)
{
fin>>x>>y;
TT[y]=x;
G[x].push_back(y);
}
fin.close();
}
void DFS(int nod)
{
C[++L]=nod;
if(v[nod])
A[nod]=A[C[L-v[nod]]]+1;
for(size_t i=0;i<G[nod].size();i++)
DFS(G[nod][i]);
C[L--]=0;
}
int TATA()
{
for(int i=1;i<=n;i++)
if(!TT[i])
return i;
return 0;
}
void print()
{
ofstream fout("cerere.out");
for(int i=1;i<=n;i++)
fout<<A[i]<<' ';
fout.close();
}
int main()
{
read();
DFS(TATA());
print();
return 0;
}