因为没有删边操作,可以直接拿并查集维护连通性。为啥不能直接带权并查集呢?因为还有修改点权。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 30010
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int n,m,fa[N],c[N][2],v[N],sum[N],fat[N],q[N];
bool rev[N];
inline int find(int x){return x==fat[x]?x:fat[x]=find(fat[x]);}
inline bool isroot(int x){
return x!=c[fa[x]][0]&&x!=c[fa[x]][1];
}
inline void update(int p){
int l=c[p][0],r=c[p][1];
sum[p]=sum[l]+v[p]+sum[r];
}
inline void pushdown(int p){
if(!rev[p]) return;rev[p]=0;
int l=c[p][0],r=c[p][1];
rev[l]^=1;rev[r]^=1;swap(c[p][0],c[p][1]);
}
inline void rotate(int x){
int y=fa[x],z=fa[y],l=x==c[y][1],r=l^1;
if(!isroot(y)) c[z][y==c[z][1]]=x;
fa[c[x][r]]=y;fa[y]=x;fa[x]=z;
c[y][l]=c[x][r];c[x][r]=y;update(y);update(x);
}
inline void splay(int x){
int top=0;q[++top]=x;
for(int xx=x;!isroot(xx);xx=fa[xx]) q[++top]=fa[xx];
while(top) pushdown(q[top--]);
while(!isroot(x)){
int y=fa[x],z=fa[y];
if(!isroot(y)){
if(x==c[y][1]^y==c[z][1]) rotate(x);
rotate(y);
}rotate(x);
}
}
inline void access(int x){
int y=0;
while(x){
splay(x);c[x][1]=y;update(x);y=x;x=fa[x];
}
}
inline void makeroot(int x){
access(x);splay(x);rev[x]^=1;
}
inline void link(int x,int y){
int xx=find(x),yy=find(y);
if(xx==yy){puts("no");return;}fat[xx]=yy;
puts("yes");makeroot(x);fa[x]=y;
}
inline void change(int x,int val){
splay(x);v[x]=val;update(x);
}
inline void query(int x,int y){
if(find(x)!=find(y)){puts("impossible");return;}
makeroot(x);access(y);splay(y);
printf("%d\n",sum[y]);
}
int main(){
// freopen("a.in","r",stdin);
n=read();for(int i=1;i<=n;++i) v[i]=read(),fat[i]=i;
m=read();
while(m--){
char op[15];scanf("%s",op+1);int x=read(),y=read();
if(op[1]=='b') link(x,y);
else if(op[1]=='p') change(x,y);
else query(x,y);
}return 0;
}