BZOJ2843: 极地旅行社

LCT

题目传送门
继续划水打模板
代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 30005
#define il inline
#define iv il void
using namespace std;
struct node{
    int fa,to[2],f,sz;
}t[N];
int a[N],n,m,stk[N],tp; char s[15];
il int _read(){
    int x=0; char ch=getchar();
    while (!isdigit(ch)) ch=getchar();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x;
}
iv writec(int x){ if (x>9) writec(x/10); putchar(x%10+48); }
iv _write(int x){ writec(x),putchar('\n'); }
iv pshp(int x){ t[x].sz=t[t[x].to[0]].sz+t[t[x].to[1]].sz+a[x]; }
iv pshd(int x){
    if (t[x].f){
        int &l=t[x].to[0],&r=t[x].to[1];
        t[l].f^=1,t[r].f^=1,t[x].f=0,swap(l,r);
    }
}
il bool rt(int x){ return t[t[x].fa].to[0]!=x&&t[t[x].fa].to[1]!=x; }
iv rtt(int x){
    int y=t[x].fa,z=t[y].fa,l=(t[y].to[1]!=x);
    if (!rt(y)) t[z].to[t[z].to[1]==y]=x;
    t[x].fa=z,t[y].fa=x,t[t[x].to[l]].fa=y;
    t[y].to[l^1]=t[x].to[l],t[x].to[l]=y;
    pshp(y),pshp(x);
}
iv splay(int x){
    stk[tp=1]=x;
    for (int i=x;!rt(i);i=t[i].fa) stk[++tp]=t[i].fa;
    for (int i=tp;i;i--) pshd(stk[i]);
    while (!rt(x)){
        int y=t[x].fa,z=t[y].fa;
        if (!rt(y)) if (t[y].to[0]==x^t[z].to[0]==y) rtt(x); else rtt(y);
        rtt(x);
    }
}
iv ccss(int x){ for (int i=0;x;i=x,x=t[x].fa) splay(x),t[x].to[1]=i,pshp(x); }
iv mkrt(int x){ ccss(x),splay(x),t[x].f^=1; }
iv lnk(int x,int y){ mkrt(x),t[x].fa=y; }
iv sprt(int x,int y){ mkrt(x),ccss(y),splay(y); }
il int find(int x){ ccss(x),splay(x); while (t[x].to[0]) x=t[x].to[0]; return x;  }
int main(){
    n=_read();
    for (int i=1;i<=n;i++) a[i]=_read();
    m=_read();
    while (m--){
        scanf("%s",s); int x=_read(),y=_read();
        switch(s[0]){
            case 'b': 
                if (find(x)!=find(y)) printf("yes\n"),lnk(x,y);
                else printf("no\n"); break;
            case 'p': ccss(x),splay(x),a[x]=y,pshp(x); break;
            case 'e':
                if (find(x)!=find(y)) printf("impossible\n");
                else sprt(x,y),_write(t[y].sz); break;
        }
    }
    return 0;
}