title: ‘BZOJ 2631 tree’
categories: BZOJ
date: 2016-1-14 22:32:00
tags: [LCT]
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.txt
3 2
1 2
2 3
* 1 3 4
\/ 1 1
output.txt
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
Solution
本题主要是tag的维护比较麻烦。具体的操作可以看代码。
有一个坑点是要开unsigned int,据黄学长说开long long会超时,于是开unsigned int就好了。。。我被这个坑了一晚上没发现,以为自己模板打挂了,好蛋疼啊QAQ……
Code
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define maxn 100000+5
#define mod 51061
using namespace std;
typedef unsigned int ll;
struct Splay_Tree{
int ch[2];
int flip,sz,pf,fa;
ll val,pl,mul,tag,sum;
}tr[maxn];
int stack[maxn];
int n,m,top;
inline int in(){
int x=0;
char ch=getchar();
while(ch<'0' || ch>'9') ch=getchar();
while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
return x;
}
void Calc(int k,int pl,int mul){
tr[k].sum=(tr[k].sum*mul+tr[k].sz*pl)%mod;
tr[k].val=(tr[k].val*mul+pl)%mod;
tr[k].pl=(tr[k].pl*mul+pl)%mod;
tr[k].mul=(tr[k].mul*mul)%mod;
}
void Pushdown(int k){
int l=tr[k].ch[0],r=tr[k].ch[1];
if(tr[k].flip){
tr[k].flip^=1;
tr[l].flip^=1; tr[r].flip^=1;
swap(tr[k].ch[0],tr[k].ch[1]);
}
if(tr[k].pl || tr[k].mul!=1){
if(l) Calc(l,tr[k].pl,tr[k].mul);
if(r) Calc(r,tr[k].pl,tr[k].mul);
tr[k].pl=0; tr[k].mul=1;
}
}
void Update(int k){
int l=tr[k].ch[0],r=tr[k].ch[1];
tr[k].sz=(tr[l].sz+tr[r].sz+1)%mod;
tr[k].sum=(tr[l].sum+tr[r].sum+tr[k].val)%mod;
}
void Rotate(int &k,int d){
int t=tr[k].ch[d^1];
tr[t].pf=tr[k].pf; tr[t].fa=tr[k].fa;
if(tr[t].ch[d]) tr[tr[t].ch[d]].fa=k;
tr[k].pf=0; tr[k].fa=t;
tr[k].ch[d^1]=tr[t].ch[d]; tr[t].ch[d]=k;
Update(k); Update(t); k=t;
}
void Splay(int &k,int x){
int d1=(tr[tr[k].ch[0]].sz<x?1:0),t=tr[k].ch[d1];
if(d1==1) x-=tr[tr[k].ch[0]].sz+1;
if(x){
int d2=(tr[tr[t].ch[0]].sz<x?1:0);
if(d2==1) x-=tr[tr[t].ch[0]].sz+1;
if(x){
Splay(tr[t].ch[d2],x);
if(d1==d2) Rotate(k,d1^1);
else Rotate(tr[k].ch[d1],d1);
}
Rotate(k,d1^1);
}
}
int rank(int k,int &rt){
int res=tr[tr[k].ch[0]].sz+1;
stack[++top]=k;
while(tr[k].fa){
int p=tr[k].fa;
stack[++top]=p;
if(tr[p].ch[0]!=k)
res+=tr[tr[p].ch[0]].sz+1;
k=p;
}
rt=k;
return res;
}
void Haha(int k){
int rt,x=rank(k,rt);
while(top) Pushdown(stack[top--]);
x=rank(k,rt);
Splay(rt,x);
}
void Cutright(int k){
if(tr[k].ch[1]){
tr[tr[k].ch[1]].pf=k;
tr[tr[k].ch[1]].fa=0;
tr[k].ch[1]=0;
Update(k);
}
}
void Access(int k){
Haha(k); Cutright(k);
while(tr[k].pf){
int u=tr[k].pf;
Haha(u); Cutright(u);
tr[u].ch[1]=k;
tr[k].pf=0; tr[k].fa=u;
Update(u); k=u;
}
}
void Beroot(int k){
Access(k); Haha(k);
tr[k].flip^=1;
}
int Find(int k){
Access(k); Haha(k);
while(tr[k].ch[0]) k=tr[k].ch[0];
return k;
}
void Link(int x,int y){
int rt1=Find(x),rt2=Find(y);
if(rt1==rt2) return;
Beroot(x); Access(y);
tr[x].pf=y;
}
void Cut(int k,int p){
Beroot(k); Access(p); Haha(p);
tr[p].ch[0]=tr[k].pf=tr[k].fa=0;
Update(p);
}
int main(){
n=in(); m=in();
for(int i=1;i<=n;i++)
tr[i].sz=tr[i].sum=tr[i].val=tr[i].mul=1;
for(int i=1;i<n;i++){
int x,y;
x=in(); y=in();
Link(x,y);
}
char s[1];
for(int i=1;i<=m;i++){
int u,v,x,y,c;
scanf("%s",s);
if(s[0]=='+'){
u=in(); v=in(); c=in();
Beroot(v); Access(u); Haha(u);
Calc(u,c,1);
}
else if(s[0]=='-'){
u=in(); v=in(); x=in(); y=in();
Cut(u,v); Link(x,y);
}
else if(s[0]=='*'){
u=in(); v=in(); c=in();
Beroot(v); Access(u); Haha(u);
Calc(u,0,c);
}
else{
u=in(); v=in();
Beroot(v); Access(u); Haha(u);
printf("%d\n",tr[u].sum);
}
}
return 0;
}