一道迟到的模板题。话说在思路清晰的时候敲代码真的爽啊。
修改操作可以先访问该节点,然后把它旋转到根,这样只用修改这一个点的权值就好了。
询问操作可以把其中一个节点搞成根,然后两点间的权值就变成了另一个节点到根的权值了。
这两个想法挺妙。
附上AC代码:
#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;
const int N=3e5+10;
int n,m,w[N],sum[N],x,y,o,mk[N],ch[N][2],f[N];
inline char nc(void){
static char ch[100010],*p1=ch,*p2=ch;
return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &a){
static char c=nc();int f=1;
for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
return (void)(a*=f);
}
#define isrt(x) ((x)!=ch[f[x]][0]&&(x)!=ch[f[x]][1])
inline void updata(int x){sum[x]=sum[ch[x][0]]^sum[ch[x][1]]^w[x];}
inline void rotate(int x){
int y=f[x],op=(x==ch[y][1]);
ch[y][op]=ch[x][op^1];
if (ch[x][op^1]) f[ch[x][op^1]]=y;
f[x]=f[y];
if (!isrt(y)) ch[f[y]][ch[f[y]][1]==y]=x;
f[y]=x,ch[x][op^1]=y;
return updata(y),updata(x);
}
inline void push(int x){if (mk[x]) mk[x]=0,mk[ch[x][0]]^=1,mk[ch[x][1]]^=1,swap(ch[x][0],ch[x][1]);}
inline void splay(int x){
static int top,sk[N];sk[top=1]=x;
for (int i=x; !isrt(i); i=f[i]) sk[++top]=f[i];
while (top) push(sk[top--]);
for (int fa=f[x]; !isrt(x); rotate(x),fa=f[x])
if (!isrt(fa)) rotate((x==ch[fa][0])==(fa==ch[f[fa]][0])?fa:x);
}
inline void access(int x){for (int lst=0; x; lst=x,x=f[x]) splay(x),ch[x][1]=lst,updata(x);}
inline void root(int x){access(x),splay(x),mk[x]^=1;}
inline void add(int x,int y){root(x),f[x]=y;}
inline void del(int x,int y){root(x),access(y),splay(y),f[x]=ch[y][0]=0;}
inline int gf(int x){access(x),splay(x); while (ch[x][0]) x=ch[x][0]; return x;}
int main(void){
read(n),read(m);
for (int i=1; i<=n; ++i) read(w[i]),sum[i]=w[i];
while (m--){
read(o),read(x),read(y);
switch (o){
case 0: root(x),access(y),splay(y),printf("%d\n",sum[y]); break;
case 1: if (gf(x)!=gf(y)) add(x,y); break;
case 2: if (gf(x)==gf(y)) del(x,y); break;
case 3: access(x),splay(x),w[x]=y,updata(x); break;
}
}
return 0;
}