Pagini recente » Cod sursa (job #1009446) | Cod sursa (job #3122649) | Istoria paginii utilizator/barosanunumberone | Cod sursa (job #2088889) | Cod sursa (job #863753)
Cod sursa(job #863753)
#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 100010
using namespace std;
int n;
int v[NMAX],sol[NMAX];
bool viz[NMAX];
vector <int> G[NMAX];
vector <int> Q;
void DFS(int nod)
{
viz[nod]=true;
Q.push_back(nod);
if(v[nod] ==0)sol[nod]=0;
else
sol[nod]=sol[Q[Q.size()-1-v[nod]]]+1;
for(int i=0;i<G[nod].size();i++)
if(!viz[G[nod][i]]) DFS(G[nod][i]);
Q.pop_back();
}
int main()
{
ifstream fin("cerere.in");
ofstream fout("cerere.out");
fin>>n;
for(int i=1;i<=n;i++)
fin>>v[i];
int t,f;
for(int i=1;i<=n-1;i++)
{
fin>>t>>f;
G[t].push_back(f);
}
sol[0]=-1;
DFS(1);
for(int i=1;i<=n;i++)
fout<<sol[i]<<" ";
return 0;
}