Pagini recente » Cod sursa (job #239411) | Cod sursa (job #3134002) | Cod sursa (job #933094) | Cod sursa (job #2356354) | Cod sursa (job #2116061)
#include<fstream>
#include<iostream>
#include<vector>
#include<cmath>
#include<queue>
#define DN 100005
#define pb push_back
using namespace std;
ifstream fin("cerere.in");
ofstream fout("cerere.out");
int n,p,q,dp[DN],k[DN],pr[25][DN],d,f;
vector<int>v[DN];
void dfs(int nod)
{
if(k[nod]!=0)
{
f=k[nod];
d=nod;
for(int i=20;i>=0&&f;i--)
if((1<<i)<=f)
{
d=pr[i][d];
f-=(1<<i);
}
dp[nod]=dp[d]+1;
}
for(auto i:v[nod])
dfs(i);
}
int main()
{
fin>>n;
for(int i=1;i<=n;i++)
fin>>k[i];
for(int i=1;i<n;i++)
{
fin>>p>>q;
pr[0][q]=p;
v[p].pb(q);
}
for(int j=1;j<25;j++)
for(int i=1;i<=n;i++)
pr[j][i]=pr[j-1][pr[j-1][i]];
for(int i=1;i<=n;i++)
if(pr[0][i]==0)
{
dfs(i);
break;
}
for(int i=1;i<=n;i++)
fout<<dp[i]<<' ';
}