[bzoj2959] 长跑
用LCT维护双联通分量。
手贱把pushnow写rotate上,,,我还能说什么
#include <bits/stdc++.h>
using namespace std;
const int N=2e5+5;
int par[N];
inline int find(int x){
return par[x]==x?par[x]:par[x]=find(par[x]);
}
struct Splay{
int ch[2],fa,sonsz,sz;bool rev;
}t[N];
int cnt;
int newnode(int _val){++cnt;t[cnt].sonsz=t[cnt].sz=_val;return cnt;}
bool son(int x){return t[t[x].fa].ch[1]==x;}
bool isroot(int x){return t[t[x].fa].ch[0]!=x&&t[t[x].fa].ch[1]!=x;}
inline void pushdown(int x){
if(t[x].rev){
swap(t[x].ch[1],t[x].ch[0]);
if(t[x].ch[1])t[t[x].ch[1]].rev^=1;
if(t[x].ch[0])t[t[x].ch[0]].rev^=1;
t[x].rev=0;
}
}
inline void pushnow(int x){
if(!isroot(x))pushnow(t[x].fa);
else t[x].fa=find(t[x].fa);
pushdown(x);
}
inline void pushup(int x){
t[x].sonsz=t[x].sz;
if(t[x].ch[0])t[x].sonsz+=t[t[x].ch[0]].sonsz;
if(t[x].ch[1])t[x].sonsz+=t[t[x].ch[1]].sonsz;
}
inline void rotate(int x){
int f=t[x].fa,g=t[t[x].fa].fa;
bool a=son(x),b=son(x)^1;
if(!isroot(f))t[g].ch[son(f)]=x;
t[x].fa=g;
t[t[x].ch[b]].fa=f;t[f].ch[a]=t[x].ch[b];
t[x].ch[b]=f;t[f].fa=x;
pushup(f);pushup(x);
}
inline void splay(int x){
pushnow(x);
while(!isroot(x)){
int f=t[x].fa;
if(!isroot(f)){
if(son(x)^son(f))rotate(x);
else rotate(f);
}
rotate(x);
}
}
void access(int x){
int tmp=0;
do{
splay(x);
t[x].ch[1]=tmp;
pushup(x);
tmp=x;x=t[x].fa;
}while(x);
}
inline void makeroot(int x){
access(x);
splay(x);
t[x].rev^=1;
}
inline void link(int x,int y){
makeroot(x);
access(y);
t[y].ch[1]=x,t[x].fa=y;
pushup(y);
splay(y);
}
void dfs(int x,int tar){
if(t[x].ch[0]){
dfs(t[x].ch[0],tar);
par[t[x].ch[0]]=tar;
}if(t[x].ch[1]){
dfs(t[x].ch[1],tar);
par[t[x].ch[1]]=tar;
}
}
inline void operate(int x,int y){//保证存在
makeroot(x);
access(y);
splay(x);
dfs(x,x);
t[x].ch[0]=0;
t[x].sz=t[x].sonsz;
pushup(x);
}
int n,m;
int par_link[N];
inline int find_link(int x){
return par_link[x]==x?x:par_link[x]=find_link(par_link[x]);
}
inline int read(){
char x=0;int res=0;
while(x<'0'||x>'9')x=getchar();
while(x<='9'&&x>='0'){res=res*10+x-'0';x=getchar();}
return res;
}
int val[N];
int main()
{
n=read(),m=read();
for(int i=1;i<=n;i++){
par[i]=par_link[i]=i;
}
for(int i=1;i<=n;i++){
val[i]=read();
newnode(val[i]);
}
while(m--){
int u,v,opt=0;
opt=read();u=read();v=read();
if(opt==1){
int uu=find(u),vv=find(v);
int x=find_link(u),y=find_link(v);
if(x^y){
par_link[x]=y;
link(uu,vv);
}else{
operate(uu,vv);
}
}else if(opt==2){
int uu=find(u);
splay(uu);
t[uu].sz+=v-val[u];
val[u]=v;
pushup(uu);
}else{
int uu=find(u),vv=find(v);
int x=find_link(u),y=find_link(v);
if(x^y){
puts("-1");
continue;
}else{
makeroot(uu);
access(vv);
splay(vv);
printf("%d\n",t[vv].sonsz);
}
}
}
}