BZOJ 2631: tree LCT

2631: tree

Time Limit: 30 Sec  Memory Limit: 128 MB
Submit: 4855  Solved: 1659
[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练手题

pushdown写错调一上午。

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

typedef unsigned int uint;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100,mod=51061;

int ch[N][2],size[N],fa[N];
uint V[N],sum[N],add[N],mul[N];
bool rev[N];

inline void pushup(int k)
{
	int ls=ch[k][0],rs=ch[k][1];
	size[k]=size[ls]+size[rs]+1;
	sum[k]=(sum[ls]+sum[rs]+V[k])%mod;
}

inline void pushdown(int k)
{
	int ls=ch[k][0],rs=ch[k][1];
	if(mul[k]!=1)
	{
		(V[ls]*=mul[k])%=mod;
		(V[rs]*=mul[k])%=mod;
		(mul[ls]*=mul[k])%=mod;
		(mul[rs]*=mul[k])%=mod;
		(add[ls]*=mul[k])%=mod;
		(add[rs]*=mul[k])%=mod;
		(sum[ls]*=mul[k])%=mod;
		(sum[rs]*=mul[k])%=mod;
		mul[k]=1;
	}
	if(add[k])
	{
		sum[ls]=(sum[ls]+add[k]*size[ls])%mod;
		sum[rs]=(sum[rs]+add[k]*size[rs])%mod;
		(add[ls]+=add[k])%=mod;
		(add[rs]+=add[k])%=mod;
		(V[ls]+=add[k])%=mod;
		(V[rs]+=add[k])%=mod;
		add[k]=0;
	}
	if(rev[k])
	{
		rev[k]=0;
		rev[ls]^=1;rev[rs]^=1;
		swap(ch[k][0],ch[k][1]);
	}
}

inline bool isroot(int k)
{return ch[fa[k]][0]!=k&&ch[fa[k]][1]!=k;}

void getdown(int k)
{
	if(!isroot(k)) getdown(fa[k]);
	pushdown(k);
}

inline void rotate(int x)
{
	int y=fa[x],z=fa[y],l,r;
	l=(ch[y][1]==x);r=l^1;
	if(!isroot(y)) ch[z][ch[z][1]==y]=x;
	fa[x]=z;fa[y]=x;fa[ch[x][r]]=y;
	ch[y][l]=ch[x][r];ch[x][r]=y;
	pushup(y);pushup(x);
}

void splay(int x)
{
	getdown(x);
	int y,z;
	while(!isroot(x))
	{
		y=fa[x];z=fa[y];
		if(!isroot(y))
		{
			if((ch[y][0]==x)^(ch[z][0]==y))rotate(x);
			else rotate(y);
		}
		rotate(x);
	}
}

void access(int x)
{int t=0;while(x){splay(x);ch[x][1]=t;pushup(x);t=x;x=fa[x];}}

void rever(int x)
{access(x);splay(x);rev[x]^=1;}

void split(int u,int v)
{rever(u);access(v);splay(v);}

void link(int u,int v)
{rever(u);fa[u]=v;splay(u);}

void cut(int u,int v)
{split(u,v);ch[v][0]=fa[u]=0;}

void modify_add(int u,int v,int val)
{
	split(v,u);
	sum[u]=(sum[u]+size[u]*val)%mod;
	(add[u]+=val)%=mod;
	(V[u]+=val)%=mod;
}

void modify_mul(int u,int v,int val)
{
	split(v,u);
	(sum[u]*=val)%=mod;
	(mul[u]*=val)%=mod;
	(add[u]*=val)%=mod;
	(V[u]*=val)%=mod;
}

void query(int u,int v)
{
	split(v,u);
	print(sum[u]);puts("");
}

int main()
{
	register int n=read(),Q=read();
	register int u,v,x,y;
	for(u=1;u<=n;++u)
		mul[u]=V[u]=sum[u]=size[u]=1;
	for(x=1;x<n;++x)
	{
		u=read();v=read();
		link(u,v);
	}
	char opt[2];
	while(Q--)
	{
		scanf("%s",opt);
		u=read();v=read();
		switch(opt[0])
		{
			case '+':
				x=read();
				modify_add(u,v,x);
				break;
			case '-':
				x=read();y=read();
				cut(u,v);link(x,y);
				break;
			case '*':
				x=read();
				modify_mul(u,v,x);
				break;
			case '/':
				query(u,v);
				break;
		}
	}
	return 0;
}