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]