【bzoj2843】极地旅行社

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行即命令，为题目描述所示。
1<=N<=30000,1<=M<=100000
Output

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

Sample Input

5

4 2 4 5 6

10

excursion 1 1

excursion 1 2

bridge 1 2

excursion 1 2

bridge 3 4

bridge 3 5

excursion 4 5

bridge 1 3

excursion 2 4

excursion 2 5
Sample Output

4

impossible

yes

6

yes

yes

15

yes

15

16

题解
lct模板。

代码

#include
#define ll long long
#define inf 1000000000
#define pa pair
using namespace std;
const int mod=1000000007;
const int N=30005;
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 f[N],c[N][2],sum[N],a[N],fa[N];
int n,m,rev[N],q[N];
int find(int i){if (f[i]!=i)f[i]=find(f[i]);return f[i];}
bool isroot(int x){return c[fa[x]][0]!=x&&c[fa[x]][1]!=x;}
inline void pushdowm(int x)
{
    int l=c[x][0],r=c[x][1];
    if (rev[x])
    {
        rev[x]^=1;rev[l]^=1;rev[r]^=1;
        swap(c[x][0],c[x][1]);
    }
}
inline void update(int x)
{
    int l=c[x][0],r=c[x][1];
    sum[x]=a[x]+sum[l]+sum[r];
}
inline void rotate(int x)
{
    int y=fa[x],z=fa[y],l,r;
    if (c[y][0]==x)l=0;else l=1;r=l^1;
    if (!isroot(y))
    {
        if (c[z][0]==y) c[z][0]=x;else c[z][1]=x;
    }
    fa[x]=z;fa[y]=x;c[y][l]=c[x][r];fa[c[x][r]]=y;c[x][r]=y;
    update(y);update(x);
}
inline void splay(int x)
{
    int top=1;q[top]=x;
    for (int i=x;!isroot(i);i=fa[i]) q[++top]=fa[i];
    while (top) pushdowm(q[top--]);
    while (!isroot(x))
    {
        int y=fa[x],z=fa[y];
        if (!isroot(y))
        {
            if (c[z][0]==y^c[y][0]==x) rotate(x);else rotate(y);
        }
        rotate(x);
    }
}
inline void access(int x)
{
    int t=0;
    while (x)
    {
        splay(x);c[x][1]=t;
        update(x);
        t=x;x=fa[x];
    }
}
inline void makeroot(int x)
{
    access(x);splay(x);rev[x]^=1;
}
inline void link(int x,int y)
{
    makeroot(x);fa[x]=y;
}
int query(int x,int y)
{
    makeroot(x);access(y);splay(y);return sum[y];
}
int main()
{
    n=read();for (int i=1;i<=n;i++) a[i]=sum[i]=read(),f[i]=i;
    m=read();
    while (m--)
    {
        char ch[20];scanf("%s",ch);int x=read(),y=read();
        if (ch[0]=='e')
        {
            if (find(x)!=find(y)) puts("impossible");
            else printf("%d\n",query(x,y));
        }
        else if (ch[0]=='b')
        {
            if (find(x)==find(y)) puts("no");
            else
            {
                puts("yes");
                link(x,y);
                f[find(x)]=find(y);
            }
        }
        else
        {
            access(x);splay(x);a[x]=y;update(x);
        }
    }
    return 0;
}