COGS[2701] 动态树

最新推荐文章于 2018-08-18 20:21:00 发布

weixin_30767921

最新推荐文章于 2018-08-18 20:21:00 发布

阅读量83

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/qjs12/p/8898246.html

本文介绍了一种使用动态树结构处理图论中特定查询问题的方法。通过实现一系列操作，如访问(access)、设置根节点(make_root)、链接(Link)及查询(query)，能够高效地更新和查询树结构中的信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

只要我们统计出一个点出发所有虚边连接的点的子树大小 x，那么对于每一个查询，先执行access，之后 x+1 即为答案。

维护方式见网上，我只提供我的板子。

// q.c

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int M=200000+10;
/*******************************************************************************************/
int f[M],l[M],r[M],s[M],sv[M]; bool tag[M];
bool is_root(int x) {
	return (l[f[x]]!=x)&&(r[f[x]]!=x);
}
void UP(int x) {
	s[x]=s[l[x]]+s[r[x]]+sv[x]+1;
}
void l_rot(int x) {
	int y,z; y=f[x],z=f[y],f[x]=z;
	if(y==l[z]) l[z]=x;
	if(y==r[z]) r[z]=x;
	if(l[x]) f[l[x]]=y;
	r[y]=l[x],l[x]=y,f[y]=x;
	UP(y),UP(x);
}
void r_rot(int x) {
	int y,z; y=f[x],z=f[y],f[x]=z;
	if(y==l[z]) l[z]=x;
	if(y==r[z]) r[z]=x;
	if(r[x]) f[r[x]]=y;
	l[y]=r[x],r[x]=y,f[y]=x;
	UP(y),UP(x);
}
void DOWN(int x) {
	if(!tag[x]) return ;
	swap(l[x],r[x]),tag[x]^=1;
	tag[l[x]]^=1,tag[r[x]]^=1;
}
void splay(int x) {
	DOWN(x); int y,z; y=f[x],z=f[y];
	while(!is_root(x)) {
		DOWN(z),DOWN(y),DOWN(x);
		if(is_root(y)) x==l[y]?r_rot(x):l_rot(x);
		else if(x==l[y]&&y==l[z]) r_rot(y),r_rot(x);
		else if(x==l[y]&&y==r[z]) r_rot(x),l_rot(x);
		else if(x==r[y]&&y==l[z]) l_rot(x),r_rot(x);
		else l_rot(y),l_rot(x); y=f[x],z=f[y]; // wtf.
	}
}
/*******************************************************************************************/
void access(int x) {
	for(int y=0;x;y=x,x=f[x]) {
		splay(x);
		sv[x]+=s[r[x]]-s[y];
		r[x]=y,UP(x); // 这里要UP.
	}
}
void make_root(int x) {
	access(x),splay(x); tag[x]^=1;
}
void Link(int x,int y) {
	make_root(x);
	access(y),splay(y); f[x]=y,sv[y]+=s[x];
}
void query(int x) {
	access(x),splay(x); printf("%d\n",sv[x]+1);
}
/*******************************************************************************************/
int main() {
	freopen("dynamic_tree.in","r",stdin);
	freopen("dynamic_tree.out","w",stdout);
	int n,m,opt,a,b;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++) {
		scanf("%d%d",&opt,&a);
		if(opt==1) make_root(a);
		else if(opt==2) query(a);
		else scanf("%d",&b),Link(b,a);
	}
	return 0;
}

转载于:https://www.cnblogs.com/qjs12/p/8898246.html