关联点(倍增)(树形结构)



这个题比较简单,就是倍增。

设计状态\(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\n",ch[i][0],ch[i][1]);
	return 0;
}

posted @ 2018-10-29 11:28  风浔凌  阅读(381)  评论(0编辑  收藏  举报