Mai intai trebuie sa te autentifici.
Cod sursa(job #1897430)
Utilizator | Data | 1 martie 2017 13:48:16 | |
---|---|---|---|
Problema | Cerere | Scor | 85 |
Compilator | cpp | Status | done |
Runda | Arhiva de probleme | Marime | 0.78 kb |
#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;
}