Pagini recente » Cod sursa (job #888236) | Cod sursa (job #70449) | Cod sursa (job #1091460) | Cod sursa (job #2569647) | Cod sursa (job #2208629)
#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 100005
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,i,j,a,b,k[NMAX],rz[NMAX];
vector<int> vc[NMAX],q;
void dfs(int n)
{
q.push_back(n);
if(k[n]==0) rz[n]=0;
else rz[n]=rz[q[q.size()-k[n]-1]]+1;
for(int i=0;i<vc[n].size();i++)
dfs(vc[n][i]);
q.pop_back();
}
int main() {
fin>>n;
for(i=1;i<=n;i++)
fin>>k[i];
for(i=1;i<n;i++)
{
fin>>a>>b;
vc[a].push_back(b);
rz[b]=1;
}
for(i=1;i<=n;i++)
if(rz[i]==0) a=i;
dfs(a);
for(i=1;i<=n;i++)
fout<<rz[i]<<" "; fout<<"\n";
}