「BZOJ2631」 tree - 动态树LCT

题目描述

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

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

输入格式

第一行两个整数n，q

接下来n-1行每行两个正整数u，v，描述这棵树

接下来q行，每行描述一个操作

输出格式

对于每个/对应的答案输出一行

数据范围

10%的数据保证，1<=n，q<=2000

另外15%的数据保证，1<=n，q<=5*10⁴，没有-操作，并且初始树为一条链

另外35%的数据保证，1<=n，q<=5*10⁴，没有-操作

100%的数据保证，1<=n，q<=10⁵，0<=c<=10⁴

分析

简单的LCT，就是标记下传比较麻烦。像luogu线段树模板2中一样，定义一个优先级，先乘法再加法，传加法标记时乘上乘法标记即可，具体看代码。

代码

#include <iostream>
#include <cstdio>
using namespace std;
const int N=200005,Mod=51061;
typedef long long LL;
int n,m,r[N],f[N];
int a[N],c[N][2],sz[N];
LL st[N],add[N],s[N];
LL mul[N],v[N];
//Splay基本操作 
bool nroot(int x) {
	return c[f[x]][0]==x||c[f[x]][1]==x;
}
void Pushup(int x) {
	s[x]=(s[c[x][0]]+s[c[x][1]]+v[x])%Mod;
	sz[x]=sz[c[x][0]]+sz[c[x][1]]+1;
}
void Pushdown(int x) {
	int &lc=c[x][0],&rc=c[x][1];
	if (r[x]&&x) {
		r[x]=0;
		if (c[x][0]) {
			swap(c[lc][0],c[lc][1]);
			r[lc]^=1;
		}
		if (c[x][1]) {
			swap(c[rc][0],c[rc][1]);
			r[rc]^=1;
		}
	}
	if (mul[x]!=1) {
		s[lc]*=mul[x];s[lc]%=Mod;
		v[lc]*=mul[x];v[lc]%=Mod;
		mul[lc]*=mul[x];mul[lc]%=Mod;
		add[lc]*=mul[x];add[lc]%=Mod;//下传时乘上乘法标记 
		s[rc]*=mul[x];s[rc]%=Mod;
		v[rc]*=mul[x];v[rc]%=Mod;
		mul[rc]*=mul[x];mul[rc]%=Mod;
		add[rc]*=mul[x];add[rc]%=Mod;
		mul[x]=1;
	}
	if (add[x]) {
		s[lc]+=sz[lc]*add[x];s[lc]%=Mod;//常规加 
		v[lc]+=add[x];v[lc]%=Mod;
		add[lc]+=add[x];add[lc]%=Mod;
		s[rc]+=sz[rc]*add[x];s[rc]%=Mod;
		v[rc]+=add[x];v[rc]%=Mod;
		add[rc]+=add[x];add[rc]%=Mod;
		add[x]=0;
	}
}
void Rotate(int x) {
	int y=f[x],z=f[y];
	int k=(c[y][1]==x),kk=(c[z][1]==y);
	if (nroot(y)) c[z][kk]=x;
	f[y]=x;
	c[y][k]=c[x][k^1];
	if (c[x][k^1]) f[c[x][k^1]]=y;
	c[x][k^1]=y;
	f[x]=z;
	Pushup(x);
	Pushup(y);
}
void Splay(int x) {
	int y=x,z=0;
	st[++z]=x;
	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)?x:y);
		Rotate(x);
	}
	Pushup(x);
}
//LCT基本操作 
void Access(int x) {
	for (int y=0;x;x=f[y=x]) {
		Splay(x);
		c[x][1]=y;
		Pushup(x);
	}
}
void Makeroot(int x) {
	Access(x);
	Splay(x);
	swap(c[x][0],c[x][1]);
	r[x]^=1;
}
void Split(int x,int y) {
	Makeroot(x);
	Access(y);
	Splay(y);
}
void Link(int x,int y) {
	Makeroot(x);
	f[x]=y;
}
void Cut(int x,int y) {
	Split(x,y);
	f[x]=c[y][0]=0;
}
void Add(int x,int y,int z) {
	Split(x,y);
	add[y]+=z;add[y]%=Mod;//直接加即可 
	v[y]+=z;v[y]%=Mod;
	s[y]+=sz[y]*z;s[y]%=Mod;
	Pushdown(y);
}
void Mul(int x,int y,int z) {
	Split(x,y);
	mul[y]*=z;mul[y]%=Mod;
	add[y]*=z;add[y]%=Mod;//同时加法标记乘上乘法标记 
	v[y]*=z;v[y]%=Mod;
	s[y]*=z;s[y]%=Mod;
	Pushdown(y);
}
void Ask(int x,int y) {
	Split(x,y);
	printf("%lld\n",s[y]);
}
int main() {
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++) v[i]=mul[i]=1;
	for (int i=1;i<n;i++) {
		int u,v;
		scanf("%d%d",&u,&v);
		Link(u,v);
	}
	for (int i=1;i<=m;i++) {
		char op[10];
		int x,y,z,w;
		scanf("%s%d%d",op,&x,&y);
		if (op[0]=='+') {
			scanf("%d",&z);
			Add(x,y,z);
		} else if (op[0]=='-') {
			scanf("%d%d",&z,&w);
			Cut(x,y);
			Link(z,w);
		} else if (op[0]=='*') {
			scanf("%d",&z);
			Mul(x,y,z);
		} else {
			Ask(x,y);
		}
	}
	return 0;
}