[洛谷P1501] [国家集训队]Tree II(LCT模板)

传送门

这是一道LCT的板子题,说白了就是在LCT上支持线段树2的操作。

所以我只是来存一个板子,并不会讲什么(再说我也不会,只能误人子弟2333)。

不过代码里的注释可以参考一下。

Code

#include<bits/stdc++.h>
using namespace std;
typedef unsigned int uint;
const int N=1e5+;
const uint mod=;
inline int read(){
int x=,w=;char ch=;
while(!isdigit(ch)) w|=ch=='-',ch=getchar();
while(isdigit(ch)) x=(x<<)+(x<<)+(ch^),ch=getchar();
return w?-x:x;
}
int f[N],sz[N],c[N][];
uint v[N],s[N],ml[N],ad[N];//int是会爆的
bool rv[N];
#define lc c[x][0]
#define rc c[x][1]
#define mul(x) x*=val,x%=mod
#define add(x) x+=val,x%=mod
//我习惯的写法是判断 not root
inline bool nrt(int x){return c[f[x]][]==x||c[f[x]][]==x;};
void pushup(int x){
s[x]=(s[lc]+s[rc]+v[x])%mod;
sz[x]=sz[lc]+sz[rc]+;
}
//自定义的优先级:乘法>加法>翻转
void Rev(int x){lc^=rc^=lc^=rc;rv[x]^=;};
void Mul(int x,uint val){mul(v[x]),mul(s[x]),mul(ml[x]),mul(ad[x]);}
void Add(int x,uint val){add(v[x]);add(ad[x]);val*=sz[x];val%=mod;add(s[x]);}
void pushdown(int x){
if(ml[x]^) Mul(lc,ml[x]),Mul(rc,ml[x]),ml[x]=;
if(ad[x]) Add(lc,ad[x]),Add(rc,ad[x]),ad[x]=;
if(rv[x]) Rev(lc),Rev(rc),rv[x]=;
}
//以下跟普通的LCT没两样
int get(int x){return x==c[f[x]][];}
void link(int x,int y,int d){c[x][d]=y;f[y]=x;}
void rotate(int x){
int y=f[x],z=f[y],d=get(x);
if(nrt(y)) c[z][get(y)]=x;f[x]=z;
//如果y=rt,说明y->z是一条虚边,也就是说x和z分属两棵不同的Splay,如果这样还连边z->x的话,后果emmm……
//但x->z必须连,因为就算y是根,把x旋上去后x就成根了,而LCT中一棵Spaly的根的父边一定是一条虚边(原树的根所属的Splay除外),相当于x继承了y连虚边的使命。。。
link(y,c[x][d^],d);
link(x,y,d^);
pushup(y),pushup(x);
}
int st[N],tp;
void splay(int x){
int t=x;
//手动用栈来pushdown
st[tp=]=t;
while(nrt(t)) st[++tp]=t=f[t];
while(tp) pushdown(st[tp--]);
for(;nrt(x);rotate(x)){
int y=f[x];
if(nrt(y)) get(x)^get(y)?rotate(x):rotate(y);
}
}
void access(int x){
for(int y=;x;x=f[y=x])
splay(x),c[x][]=y,pushup(x);
}
void makert(int x){
access(x),splay(x),Rev(x);
}
int findrt(int x){
access(x),splay(x);
while(lc) pushdown(x),x=lc;
splay(x);return x;
}
void split(int x,int y){
makert(x),access(y),splay(y);
}
void link(int x,int y){
makert(x);if(findrt(y)^x) f[x]=y;
}
void cut(int x,int y){
makert(x);
//在这道题中由于保证了cut操作合法因此应该可以不加判断
if(findrt(y)==x&&f[y]==x&&!c[y][]) f[y]=c[x][]=,pushup(x);
}
int n,m;
int main(){
n=read(),m=read();
for(int i=;i<=n;++i) v[i]=ml[i]=sz[i]=;
for(int i=;i<n;++i) link(read(),read());
char op[];int x,y;
while(m--){
scanf("%s",op);
x=read(),y=read();
switch(op[]){
case '+':split(x,y);Add(y,read());break;
case '-':cut(x,y);link(read(),read());break;
case '*':split(x,y);Mul(y,read());break;
case '/':split(x,y);cout<<s[y]<<endl;break;
}
}
return ;
}

LCT模板

上一篇:SPSS Clementine 数据挖掘入门2


下一篇:BZOJ3282: Tree (LCT模板)