BZOJ2631: tree Link-cut-tree

BZOJ2631: tree

Time Limit: 30 Sec   Memory Limit: 128 MB

Submit: 3717   Solved: 1238

[Submit][Status][Discuss]

题解：

LCT，同时维护乘法和加法，在推乘法标记的时候注意先要更新加法标记

同时维护子树的和要注意必须时时更新不能等着Pushup回来

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define int unsigned int
const int N=300005;
const int MOD=51061;
int n,m,ls[N],rs[N],a[N],rev[N],sum[N],siz[N],mx[N],fa[N],w[N],plu[N],mul[N];
char o[9];
void Pushup(int k)
{
	sum[k]=(sum[ls[k]]+sum[rs[k]]+a[k])%MOD;
	siz[k]=siz[ls[k]]+siz[rs[k]]+1;
}
void Pudrev(int k)
{
	if(!k) return;
	swap(ls[k],rs[k]);
	rev[k]^=1;
}
void Pudmul(int k,int v)
{
	if(!k) return;
	a[k]=(a[k]*v)%MOD,sum[k]=(sum[k]*v)%MOD;
	mul[k]=(mul[k]*v)%MOD,plu[k]=(plu[k]*v)%MOD;
}
void Pudplu(int k,int v)
{
	if(!k) return;
	a[k]=(a[k]+v)%MOD,sum[k]=(sum[k]+siz[k]*v)%MOD;
	plu[k]=(plu[k]+v)%MOD;
}
void Pushdown(int k)
{
	if(rev[k])
	{
		Pudrev(ls[k]);
		Pudrev(rs[k]);
		rev[k]=0;
	}
	if(mul[k]!=1)
	{
		Pudmul(ls[k],mul[k]);
		Pudmul(rs[k],mul[k]);
		mul[k]=1;
	}
	if(plu[k])
	{
		Pudplu(ls[k],plu[k]);
		Pudplu(rs[k],plu[k]);
		plu[k]=0;
	}
}
bool Check(int k)
{
	return ls[fa[k]]!=k&&rs[fa[k]]!=k;
}
void Rotate(int k)
{
	int y=fa[k],z=fa[y];
	if(!Check(y))
	{
		if(ls[z]==y) ls[z]=k;
		else rs[z]=k;
	}
	fa[k]=z;
	fa[y]=k;
	if(k==ls[y])
	{
		ls[y]=rs[k];
		fa[rs[k]]=y;
		rs[k]=y;
	}
	else
	{
		rs[y]=ls[k];
		fa[ls[k]]=y;
		ls[k]=y;
	}
	Pushup(y);
	Pushup(k);
}
void Splay(int k)
{
	int len=1;
	w[1]=k;
	for(int i=k;!Check(i);i=fa[i]) w[++len]=fa[i];
	for(int i=len;i;i--) Pushdown(w[i]);
	while(!Check(k))
	{
		int y=fa[k],z=fa[y];
		if(!Check(y))
		{
			if(ls[y]==k^ls[z]==y) Rotate(k);
			else Rotate(y);
		}
		Rotate(k);
	}
}
void Access(int k)
{
    int x=0;
    while(k)
    {
        Splay(k);
        rs[k]=x;
        Pushup(k);
        x=k;
        k=fa[k];
    }
}
void Evert(int k)
{
    Access(k);
    Splay(k);
    rev[k]^=1;
    swap(ls[k],rs[k]);
}
void EAS(int u,int v)
{
	Evert(u),Access(v),Splay(v);
}
void Link(int u,int v)
{
    Evert(u);
	fa[u]=v;
}
void Cut(int u,int v)
{
    EAS(u,v);
    fa[u]=ls[v]=0;
    Pushup(v);
}
int Find(int k)
{
    Access(k);
    Splay(k);
    while(ls[k]) k=ls[k];
    return k;
}
main()
{
	scanf("%u%u",&n,&m);
	int u,v,w,x,y;
	for(int i=1;i<=n;i++) a[i]=mul[i]=siz[i]=1;
	for(int j=1;j<n;j++)
	{
		scanf("%u%u",&u,&v);
		Link(u,v);
	}
	while(m--)
	{
		scanf("%s",o);
		if(o[0]=='+')
		{
			scanf("%u%u%u",&u,&v,&x);
			EAS(u,v);
			Pudplu(v,x);
			Pushup(v);
		}
		if(o[0]=='*')
		{
			scanf("%u%u%u",&u,&v,&x);
			EAS(u,v);
			Pudmul(v,x);
			Pushup(v);
		}
		if(o[0]=='-')
		{
			scanf("%u%u%u%u",&u,&v,&w,&x);
			Cut(u,v),Link(w,x);
		}
		if(o[0]=='/')
		{
			scanf("%u%u",&u,&v);
			EAS(u,v);
			printf("%u\n",sum[v]);
		}
	}
}

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

Source

[Submit][Status][Discuss]