Pagini recente » Cod sursa (job #583206) | Cod sursa (job #1141514) | Cod sursa (job #2458141) | Istoria paginii runda/info.conc.sept.2/clasament | Cod sursa (job #1897430)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
const int NMAX = 100005;
int n,x,y;
int kcost[NMAX];
int dp[20][NMAX];
int getParent(int nod,int wow)
{
int l=kcost[nod];
for(int p=0 ; (1<<p)<=l ; p++)
if(l&(1<<p))
nod=dp[p][nod];
if(kcost[nod]) return getParent(nod,wow+1);
return wow;
}
int main()
{
fin>>n;
for(int i=1;i<=n;i+=1) fin>>kcost[i];
for(int i=1;i<n;i+=1)
{
fin>>x>>y;
dp[0][y]=x;
}
for(int p=1; (1<<p)<=n ; p++)
for(int i=1; i<=n; i++)
dp[p][i]=dp[p-1][dp[p-1][i]];
for(int i=1;i<=n;i+=1)
{
if(kcost[i]==0) fout<<0<<' ';
else fout << getParent(i,1)<<' ';
}
return 0;
}