洛谷P3950 部落冲突(LCT)

洛谷题目传送门

最无脑LCT题解,Dalao们的各种算法都比这个好多啦。。。

唯一的好处就是只管码代码就好了

开战cut,停战link,询问findroot判连通性

太无脑,应该不用打注释了。常数大就不用说了(逃

#include<cstdio>
#include<cstdlib>
#define R register int
#define I inline void
#define lc c[x][0]
#define rc c[x][1]
#define G ch=getchar()
#define gc G;while(ch<'-')G
#define in(z) gc;z=ch&15;G;while(ch>'-')z*=10,z+=ch&15,G;
const int N=300009;
int f[N],c[N][2],st[N],u[N],v[N];
bool r[N];
inline bool nroot(R x){
	return c[f[x]][0]==x||c[f[x]][1]==x;
}
I pushdown(R x){
	if(r[x]){
		R t=lc;lc=rc;rc=t;
		r[lc]^=1,r[rc]^=1,r[x]=0;
	}
}
I rotate(R x){
	R y=f[x],z=f[y],k=c[y][1]==x,w=c[x][!k];
	if(nroot(y))c[z][c[z][1]==y]=x;c[x][!k]=y;c[y][k]=w;
	if(w)f[w]=y;f[y]=x;f[x]=z;
}
I splay(R x){
	R y=x,z=0;
	st[++z]=y;
	while(nroot(y))st[++z]=y=f[y];
	while(z)pushdown(st[z--]);
	while(nroot(x)){
		y=f[x];z=f[y];
		if(nroot(y))
			rotate((c[y][0]==x)^(c[z][0]==y)?y:x);
		rotate(x);
	}
}
I access(R x){
	for(R y=0;x;x=f[y=x])
		splay(x),rc=y;
}
I makeroot(R x){
	access(x);splay(x);
	r[x]^=1;
}
inline int findroot(R x){
	access(x);splay(x);
    pushdown(x);
	while(lc)pushdown(x=lc);
	return x;
}
I split(R x,R y){
	makeroot(x);
	access(y);splay(y);
}
I link(R x,R y){
	makeroot(x);f[x]=y;
}
I cut(R x,R y){
	split(x,y);f[x]=c[y][0]=0;
}
int main()
{
	register char ch;
	R n,m,p=0,a,b;
	in(n);in(m);
	for(R i=1;i<n;++i){in(a);in(b);link(a,b);}
	while(m--){
		gc;
		switch(ch){
		case 'U':in(a);link(u[a],v[a]);break;
		case 'C':in(a);in(b);++p;cut(u[p]=a,v[p]=b);break;
		case 'Q':in(a);in(b);puts(findroot(a)==findroot(b)?"Yes":"No");
		}
	}
	return 0;
}
posted @ 2018-03-26 20:28  Flash_Hu  阅读(303)  评论(0编辑  收藏