这个题比较简单,就是倍增。
设计状态(f[i][j])为从节点(i)跳(2^k)步能够跳到的节点。
代码如下:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define MAXN 200010
using namespace std;
int n;
int sum[MAXN],ch[MAXN][2],l[MAXN],r[MAXN],f[MAXN][33];
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&sum[i]);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&l[i],&r[i]);
f[l[i]][0]=f[r[i]][0]=i;
}
f[0][0]=0;
for(int j=1;j<=31;j++)
for(int i=1;i<=n;i++)
f[i][j]=f[f[i][j-1]][j-1];
for(int i=1;i<=n;i++)
{
int now=i;
for(int j=0;(1<<j)<=sum[i]-1;j++)
if((sum[i]-1)&(1<<j)) now=f[now][j];
int fa=f[now][0];
if(l[fa]==now) ch[fa][0]++;
else ch[fa][1]++;
}
for(int i=1;i<=n;i++)
printf("%d %d
",ch[i][0],ch[i][1]);
return 0;
}