bzoj2631 tree

2631: tree

Time Limit: 30 Sec   Memory Limit: 128 MB
Submit: 3433   Solved: 1155
[ Submit][ Status][ Discuss]

Description

　一棵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的余数。

Input

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

Output

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

Sample Input

3 2
1 2
2 3
* 1 3 4
/ 1 1

Sample Output

4

HINT

数据规模和约定

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

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

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

100%的数据保证，1<=n，q<=10^5，0<=c<=10^4

LCT模板题

注意：一般access之后都要splay

#include
#include
#include
#include
#include
#include
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define N 100005
#define mod 51061
using namespace std;
int n,m;
int fa[N],c[N][2],sz[N],rev[N],st[N];
ll sum[N],w[N],add[N],mul[N];
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
bool isroot(int x)
{
	return c[fa[x]][0]!=x&&c[fa[x]][1]!=x;
}
void update(int x,ll ad,ll mu)//这里两种标记同时处理，很巧妙 
{
	w[x]=(w[x]*mu+ad)%mod;
	sum[x]=(sum[x]*mu+ad*sz[x])%mod;
	add[x]=(add[x]*mu+ad)%mod;
	mul[x]=mul[x]*mu%mod;
}
void pushup(int x)
{
	sum[x]=(sum[c[x][0]]+sum[c[x][1]]+w[x])%mod;
	sz[x]=sz[c[x][0]]+sz[c[x][1]]+1;
}
void pushdown(int x)
{
	if (rev[x])
	{
		int ls=c[x][0],rs=c[x][1];
		rev[x]^=1;rev[ls]^=1;rev[rs]^=1;
		swap(c[x][0],c[x][1]);
	}
	if (!add[x]&&mul[x]==1) return;
	if (c[x][0]) update(c[x][0],add[x],mul[x]);
	if (c[x][1]) update(c[x][1],add[x],mul[x]);
	add[x]=0;mul[x]=1; 
}
void rotate(int x)
{
	int y=fa[x],z=fa[y],l=c[y][0]==x?0:1,r=l^1;
	if (!isroot(y)) c[z][c[z][0]==y?0:1]=x;
	fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
	c[y][l]=c[x][r];c[x][r]=y;
	pushup(y);pushup(x);
}
void splay(int x)
{
	int top=0;st[++top]=x;
	for(int i=x;!isroot(i);i=fa[i]) st[++top]=fa[i];
	D(i,top,1) pushdown(st[i]);
	while (!isroot(x))
	{
		int y=fa[x],z=fa[y];
		if (!isroot(y))
		{
			if (c[y][0]==x^c[z][0]==y) rotate(x);
			else rotate(y);
		}
		rotate(x);
	}
}
void access(int x)
{
	int t=0;
	while (x)
	{
		splay(x);c[x][1]=t;pushup(x);
		t=x;x=fa[x];
	}
}
void moveroot(int x)
{
	access(x);splay(x);rev[x]^=1;
}
void link(int x,int y)
{
	moveroot(x);fa[x]=y;
}
void cut(int x,int y)
{
	moveroot(x);access(y);splay(y);
	c[y][0]=fa[x]=0;pushup(y);
}
void change(int x,int y,int ad,int mu)
{
	moveroot(x);access(y);splay(y);
	update(y,ad,mu);
}
ll query(int x,int y)
{
	moveroot(x);access(y);splay(y);
	return sum[y];
}
int main()
{
	n=read();m=read();
	F(i,1,n) sum[i]=w[i]=mul[i]=1,add[i]=0;
	F(i,1,n-1){int x=read(),y=read();link(x,y);}
	while (m--)
	{
		int x,y,z,w;char ch=getchar();
		while (ch!='+'&&ch!='-'&&ch!='*'&&ch!='/') ch=getchar();
		if (ch=='+'){x=read();y=read();z=read();change(x,y,z,1);}
		else if (ch=='-'){x=read();y=read();z=read();w=read();cut(x,y);link(z,w);}
		else if (ch=='*'){x=read();y=read();z=read();change(x,y,0,z);}
		else {x=read();y=read();printf("%lld\n",query(x,y));}
	}
	return 0;
}