bzoj 2843: 极地旅行社

Description

不久之前，Mirko建立了一个旅行社，名叫“极地之梦”。这家旅行社在北极附近购买了N座冰岛，并且提供观光服务。当地最受欢迎的当然是帝企鹅了，这些小家伙经常成群结队的游走在各个冰岛之间。
Mirko的旅行社遭受一次重大打击，以至于观光游轮已经不划算了。旅行社将在冰岛之间建造大桥，并用观光巴士来运载游客。Mirko希望开发一个电脑程序来管理这些大桥的建造过程，以免有不可预料的错误发生。
这些冰岛从1到N标号。一开始时这些岛屿没有大桥连接，并且所有岛上的帝企鹅数量都是知道的。每座岛上的企鹅数量虽然会有所改变，但是始终在[0, 1000]之间。
你的程序需要处理以下三种命令：
1.”bridge A B”——在A与B之间建立一座大桥（A与B是不同的岛屿）。由于经费限制，这项命令被接受，当且仅当A与B不联通。若这项命令被接受，你的程序需要输出”yes”，之后会建造这座大桥。否则，你的程序需要输出”no”。
2.”penguins A X”——根据可靠消息，岛屿A此时的帝企鹅数量变为X。这项命令只是用来提供信息的，你的程序不需要回应。
3.”excursion A B”——一个旅行团希望从A出发到B。若A与B连通，你的程序需要输出这个旅行团一路上所能看到的帝企鹅数量（包括起点A与终点B），若不联通，你的程序需要输出”impossible”。
Input

第一行一个正整数N，表示冰岛的数量。
第二行N个范围[0, 1000]的整数，为每座岛屿初始的帝企鹅数量。
第三行一个正整数M，表示命令的数量。
接下来M行即命令，为题目描述所示。
Output

对于每个bridge命令与excursion命令，输出一行，为题目描述所示。

本题就是一道LCT裸题，建桥之前查询一下是否在一棵树里，然后连边即可。询问的话将询问的点access一下，然后splay使其处于splay的根节点，输出该splay中的总企鹅数即可。
关于LCT可以查看我的另一片博文 在这里！
下面是代码：

#include<iostream>
#include<cstdio>
#define N 30010
#define M 100010
#define lc(o) ch[o][0]
#define rc(o) ch[o][1]
using namespace std;
int n,m,num[N],fa[N],sum[N],ch[N][2],re[N];
int is(int o) { return rc(fa[o])==o; }
int isroot(int o) { return !fa[o]||(lc(fa[o])!=o&&rc(fa[o])!=o); }
void markdown(int o) {
    if(re[o]) {
      re[o]=0;swap(lc(o),rc(o));
      if(lc(o)) re[lc(o)]^=1;
      if(rc(o)) re[rc(o)]^=1;
    } return;
}
void update(int o) {
    if(!o) return;sum[o]=num[o];
    if(lc(o)) sum[o]+=sum[lc(o)];
    if(rc(o)) sum[o]+=sum[rc(o)];
}
void rorate(int o) {
    int f=fa[o],d=is(o),c=ch[o][d^1];
    if(c) fa[c]=f;ch[f][d]=c;
    if(!isroot(f)) ch[fa[f]][is(f)]=o;fa[o]=fa[f];
    ch[o][d^1]=f;fa[f]=o; update(f),update(o);
}
void splay(int o) {
    if(o==0) return;
    markdown(fa[o]);markdown(o);
    while(!isroot(o)) {
      int p=fa[o];
      markdown(fa[p]);markdown(p);markdown(o);
      if(isroot(p)) rorate(o);
      else if(is(o)==is(p)) rorate(p),rorate(o);
      else rorate(o),rorate(o);
    }
}
void access(int o,int x=0) {
    for(;o;o=fa[x=o]) splay(o),rc(o)=x;
}
void makeroot(int o) {
    access(o); splay(o); re[o]^=1;
}
int findroot(int o) {
    while(!isroot(o)) o=fa[o];return o;
}
void bridge(int a,int b) {
    makeroot(b);access(a);
    if(findroot(a)==findroot(b)) printf("no\n");
    else splay(b),fa[b]=a,printf("yes\n");
}
void penguins(int a,int x) {
    splay(a);num[a]=x;update(a);
}
void excursion(int a,int b) {
    makeroot(b); access(a);
    if(findroot(a)!=findroot(b)) printf("impossible\n");
    else {
      splay(a);printf("%d\n",sum[lc(a)]+num[a]);
    } return;
}
int in() {
    int s=0;char c=getchar(); while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') s=s*10+c-'0',c=getchar();return s;
}
void input(char &s) {
    while(s<'a'||s>'z') s=getchar();char c=getchar();
    while(c>='a'&&c<='z') c=getchar(); return;
}
int main() {
    n=in();for(int i=1;i<=n;i++) sum[i]=num[i]=in();
    m=in();for(int i=1;i<=m;i++) {
      char s=0;input(s);int a=in(),b=in();
      if(s=='e') excursion(a,b);
      else if(s=='b') bridge(a,b);
      else penguins(a,b);
    }
}