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);    }}