【BZOJ 2631】tree

传送门

---

Problem

一棵 $n$ 个点的树，每个点的初始权值为 $1$。对于这棵树有 $q$ 个操作，每个操作为以下四种操作之一：

• $+$ $u$ $v$ $c$：将 $u$ 到 $v$ 的路径上的点的权值都加上自然数 $c$；
• $-$ $u_1$ $v_1$ $u_2$ $v_2$：将树中原有的边 $(u_1,v_1)$ 删除，加入一条新边 $(u_2,v_2)$，保证操作完之后仍然是一棵树；
• $\ast$ $u$ $v$ $c$：将 $u$ 到 $v$ 的路径上的点的权值都乘上自然数 $c$；
• $/$ $u$ $v$：询问 $u$ 到 $v$ 的路径上的点的权值和，求出答案对于 $51061$ 的余数。

---

Solution

建议先做一下洛谷上的线段树模板 $2$。

这道题就是把它搬到了树上，然后添加了删边和加边的操作。

就直接用 $\mathrm{L}\mathrm{C}\mathrm{T}$ 维护就可以了。

注意一下，虽然是对 $51061$ 取模，但是 $51060\times 51060=2607123600$，还是超了 $int$ 的范围。

---

Code

#include<stack>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define Mod 51061
using namespace std;
char op[5];
namespace LCT{
	int fa[N],val[N],mul[N],add[N],sum[N],mark[N],Size[N],son[N][2];
	bool Get(int x)  {return son[fa[x]][1]==x;}
	bool isroot(int x)  {return (son[fa[x]][0]!=x)&&(son[fa[x]][1]!=x);}
	void pushup(int x){
		Size[x]=Size[son[x][0]]+Size[son[x][1]]+1;
		sum[x]=(sum[son[x][0]]+sum[son[x][1]]+val[x])%Mod;
	}
	void pushmul(int x,int v){
		sum[x]=1ll*sum[x]*v%Mod;
		add[x]=1ll*add[x]*v%Mod;
		mul[x]=1ll*mul[x]*v%Mod;
		val[x]=1ll*val[x]*v%Mod;
	}
	void pushadd(int x,int v){
		sum[x]=(sum[x]+1ll*v*Size[x]%Mod)%Mod;
		add[x]=(add[x]+v)%Mod,val[x]=(val[x]+v)%Mod;
	}
	void pushflip(int x){
		swap(son[x][0],son[x][1]);
		mark[son[x][0]]^=1,mark[son[x][1]]^=1;
	}
	void pushdown(int x){
		if(mul[x]!=1)  pushmul(son[x][0],mul[x]),pushmul(son[x][1],mul[x]),mul[x]=1;
		if(add[x]!=0)  pushadd(son[x][0],add[x]),pushadd(son[x][1],add[x]),add[x]=0;
		if(mark[x]!=0)  pushflip(x),mark[x]=0;
	}
	void Rotate(int x){
		int y=fa[x],z=fa[y];
		int k=Get(x),l=son[x][k^1];
		son[y][k]=l,fa[l]=(l?y:0);
		if(!isroot(y))  son[z][Get(y)]=x;fa[x]=z;
		son[x][k^1]=y,fa[y]=x;
		pushup(y),pushup(x);
	}
	stack<int>stk;
	void Splay(int x){
		stk.push(x);
		for(int i=x;!isroot(i);i=fa[i])  stk.push(fa[i]);
		while(!stk.empty())  pushdown(stk.top()),stk.pop();
		while(!isroot(x)){
			int y=fa[x];
			if(!isroot(y))  Rotate(Get(x)==Get(y)?y:x);
			Rotate(x);
		}
	}
	void Access(int x){
		for(int i=0;x;x=fa[i=x])
			Splay(x),son[x][1]=i,pushup(x);
	}
	void Makeroot(int x){
		Access(x),Splay(x),mark[x]^=1;
	}
	void split(int x,int y){
		Makeroot(x),Access(y),Splay(y);
	}
	void Link(int x,int y){
		Makeroot(x),fa[x]=y;
	}
	void Cut(int x,int y){
		split(x,y);
		fa[x]=son[y][0]=0;
		pushup(y);
	}
}
using namespace LCT;
int main(){
	int n,m,i,x,y,p,q;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;++i)
		val[i]=Size[i]=mul[i]=1;
	for(i=1;i<n;++i){
		scanf("%d%d",&x,&y);
		Link(x,y);
	}
	for(i=1;i<=m;++i){
		scanf("%s",op);
		if(op[0]=='+'){
			scanf("%d%d%d",&x,&y,&p);
			split(x,y),pushadd(y,p);
		}
		else  if(op[0]=='-'){
			scanf("%d%d",&x,&y),Cut(x,y);
			scanf("%d%d",&p,&q),Link(p,q);
		}
		else  if(op[0]=='*'){
			scanf("%d%d%d",&x,&y,&p);
			split(x,y),pushmul(y,p);
		}
		else{
			scanf("%d%d",&x,&y),split(x,y);
			printf("%d\n",sum[y]);
		}
	}
	return 0;
}