Pagini recente » Cod sursa (job #2611839) | Cod sursa (job #917492) | Cod sursa (job #2206926) | Cod sursa (job #2089454) | Cod sursa (job #1827795)
#include <fstream>
#include <vector>
#define DIM 100001
using namespace std;
int i,n,x,y,v[DIM],sol[DIM],sol2[DIM],v2[DIM],w[DIM];
vector <int> L[DIM];
ifstream fin ("cerere.in");
ofstream fout ("cerere.out");
void dfs (int nod,int niv){
v2[nod] = 1;
if (v[nod] == 0)
sol[niv] = 0;
else
sol[niv] = sol[niv-v[nod]]+1;
sol2[nod] = sol[niv];
for (int i=0;i<L[nod].size();i++)
if (v2[L[nod][i]] == 0)
dfs (L[nod][i],niv+1);
}
int main (){
fin>>n;
for (i=1;i<=n;i++)
fin>>v[i];
for (i=1;i<n;i++){
fin>>x>>y;
L[x].push_back(y);
// L[y].push_back(x);
w[y] = 1;
}
for (i=1;i<=n;i++){
if (w[i] == 0){
dfs (i,1);
break;
}
}
for (i=1;i<=n;i++)
fout<<sol2[i]<<" ";
return 0;
}