【BZOJ 2631】tree LCT

就像是一开始的线段树维护序列，不过换到树上就只好LCT咯。

#include<cstdio>
#include<cstring>
#include<iostream>
#define maxn 100020
#define ls(a) ch[a][0]
#define rs(a) ch[a][1]
#define Mod 51061
#define add(a,b) (a=(a+(b))%Mod)
#define mul(a,b) (a=(a*(b))%Mod)
#define uint unsigned int
using namespace std;
int read(){
	char c=getchar();int f=1,x=0;
	for(;c>'9'||c<'0';c=getchar())if(c=='-')f=-1;
	for(;c>='0'&&c<='9';c=getchar())x=x*10+c-'0';
	return x*f;
}
int ch[maxn][2],fa[maxn],n,m,q[maxn],top,size[maxn],flag[maxn];
uint l1[maxn],l2[maxn],sum[maxn],val[maxn];

inline int Q(int u){return u==rs(fa[u]);}
inline bool isrt(int u){return !fa[u]||(u!=ls(fa[u])&&u!=rs(fa[u]));}
inline void push_up(int u){sum[u]=(sum[ls(u)]+sum[rs(u)]+val[u])%Mod,size[u]=(size[ls(u)]+size[rs(u)]+1)%Mod;}
inline void push_down(int u){
	if(u==0||(l1[u]==1&&l2[u]==0&&flag[u]==0))return;
	int l=ls(u),r=rs(u);
	if(flag[u]){
		swap(ls(u),rs(u));flag[u]=0;
		if(ls(u))flag[ls(u)]^=1;
		if(rs(u))flag[rs(u)]^=1;
	}
	if(l1[u]!=1){
		if(l)mul(sum[l],l1[u]),mul(l1[l],l1[u]),mul(l2[l],l1[u]),mul(val[l],l1[u]);
		if(r)mul(sum[r],l1[u]),mul(l1[r],l1[u]),mul(l2[r],l1[u]),mul(val[r],l1[u]);
		l1[u]=1;
	}
	if(l2[u]){
		if(l)add(sum[l],l2[u]*size[l]%Mod),add(l2[l],l2[u]),add(val[l],l2[u]);
		if(r)add(sum[r],l2[u]*size[r]%Mod),add(l2[r],l2[u]),add(val[r],l2[u]);
		l2[u]=0;
	}
	
}
void rotate(int u){
	if(!u)return;
	int f=fa[u],ff=fa[f],d=!Q(u);
	if(!isrt(f))ch[ff][Q(f)]=u;fa[u]=ff;
	if(ch[u][d])fa[ch[u][d]]=f;ch[f][!d]=ch[u][d];
	ch[u][d]=f,fa[f]=u;
	push_up(f),push_up(u);
}
void splay(int u){
	if(!u)return;
	top=0;q[++top]=u;
	for(int i=u;!isrt(i);i=fa[i])q[++top]=fa[i];
	for(int i=top;i>=1;i--)push_down(q[i]);
	while(!isrt(u)){
		int f=fa[u];
		if(!isrt(f)){
			if(Q(f)==Q(u))rotate(f);
			else rotate(u);
		}
		rotate(u);
	}
}
void access(int u){
	for(int x=0;u;u=fa[x=u]){
		splay(u);rs(u)=x;
		push_up(u);
	}
}
void makert(int u){
	access(u);
	splay(u);
	flag[u]^=1;
}
void link(int a,int b){
	makert(a);splay(a);fa[a]=b;
}
void cut(int a,int b){
	makert(a);access(b);splay(b);
	fa[a]=0,ch[b][0]=0;
	push_up(b);
}
void update_mul(int a,int b,int add){
	makert(a);
	access(b);
	splay(b);
	mul(sum[b],add),mul(val[b],add),mul(l1[b],add),mul(l2[b],add);
}
void update_add(int a,int b,int add){
	makert(a);access(b);splay(b);
	add(sum[b],add*size[b]%Mod),add(val[b],add),add(l2[b],add);
}
void query(int a,int b){
	makert(a);access(b);splay(b);
	printf("%u\n",sum[b]%Mod);
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++)val[i]=l1[i]=size[i]=sum[i]=1;
	for(int a,b,i=1;i<n;i++){
		a=read(),b=read();
		link(a,b);
	}char s[5];int a,b,c,d;
	while(m--){
		scanf("%s",s);a=read(),b=read();
		if(s[0]=='/')query(a,b);
		else if(s[0]=='*'){
			scanf("%d",&c);
			update_mul(a,b,c);
		}else if(s[0]=='+'){
			c=read();
			update_add(a,b,c);
		}else{
			c=read(),d=read();
			cut(a,b),link(c,d);
		}
	}
	return 0;
}