Pagini recente » Cod sursa (job #1465261) | Cod sursa (job #1854420) | Cod sursa (job #2615898) | Cod sursa (job #2820106) | Cod sursa (job #2205591)
#include <iostream>
#include <fstream>
#include <vector>
#define MAXN 100005
using namespace std;
ifstream in("cerere.in");
ofstream out("cerere.out");
vector<int>graf[MAXN];
bool viz[MAXN],e_sursa[MAXN];
int n,stramosi[MAXN],drum[MAXN],dp[MAXN],sursa,nr_noduri;
void dfs(int nod){
drum[++nr_noduri] = nod;
if(stramosi[nod])
dp[nod] = dp[drum[nr_noduri - stramosi[nod]]] + 1;
else
dp[nod] = 0;
viz[nod] = true;
int curent;
for(int i = 0; i < graf[nod].size(); i++){
curent = graf[nod][i];
if(!viz[curent])
dfs(curent);
}
--nr_noduri;
}
int main()
{
in>>n;
for(int i = 1; i <= n; i++)
in>>stramosi[i];
for(int i = 1; i < n; i++){
int tata,fiu;
in>>tata>>fiu;
e_sursa[fiu] = true;
graf[tata].push_back(fiu);
}
for(int i = 1; i <= n && !sursa; i++)
if(!e_sursa[i])
sursa = i;
dfs(sursa);
for(int i = 1; i <= n; i++)
out<<dp[i]<<" ";
return 0;
}