Pagini recente » Cod sursa (job #3204060) | Cod sursa (job #1655189) | Cod sursa (job #1225795) | Cod sursa (job #2613871) | Cod sursa (job #1950034)
#include <iostream>
#include <vector>
#include <fstream>
#define NMAX 100005
using namespace std;
ifstream in("cerere.in");
ofstream out("cerere.out");
int n,m,c[NMAX],t[NMAX],viz[NMAX],x,y,cost,d[NMAX],stiva[NMAX],index;
vector<int> a[NMAX];
void dfs(int nod) {
viz[nod] = 1;
if(c[nod] == 0) d[nod] = 0;
else d[nod] = 1+ d[ stiva[ index-c[nod] ] ];
stiva[index++] = nod;
for(int i=0;i<a[nod].size();i++) {
if(!viz[a[nod][i]]){
dfs(a[nod][i]);
index--;
}
}
}
int main()
{
in >> n;
for(int i=1;i<=n;i++) {
in >> c[i];
}
for(int i=1;i<n;i++) {
in >> x >> y;
t[y] = x;
a[x].push_back(y);
}
int rad = 1;
for(int i=1;i<=n;i++) {
if(t[i] == 0) {
rad = i;
break;
}
}
dfs(rad);
for(int i=1;i<=n;i++) {
//cout << t[i] << " ";
out << d[i] << " ";
}//
//cout << calcStr(4);
return 0;
}