BZOJ2843:极地旅行社(LCT)

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

Solution

这两天FFT做的我头疼……先做点水题换换脑子

我的Rotate函数又㕛叒叕打错T了两次真是喜闻乐见

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#define N (30000+1000)
using namespace std;

int n,m,x,y,Sum[N],Son[N][2],Father[N],Rev[N],Val[N];
char opt[30];

void Update(int x){Sum[x]=Sum[Son[x][0]]+Sum[Son[x][1]]+Val[x];}
int Get(int x){return Son[Father[x]][1]==x;}
int Is_root(int x){return Son[Father[x]][0]!=x && Son[Father[x]][1]!=x;}

void Rotate(int x)
{
    int wh=Get(x);
    int fa=Father[x],fafa=Father[fa];
    if (!Is_root(fa)) Son[fafa][Son[fafa][1]==fa]=x;
    Son[fa][wh]=Son[x][wh^1]; Father[fa]=x;
    if (Son[fa][wh]) Father[Son[fa][wh]]=fa;
    Son[x][wh^1]=fa; Father[x]=fafa;
    Update(fa); Update(x);
}

void Pushdown(int x)
{
    if (Rev[x] && x)
    {
        if (Son[x][0]) Rev[Son[x][0]]^=1;
        if (Son[x][1]) Rev[Son[x][1]]^=1;
        swap(Son[x][0],Son[x][1]);
        Rev[x]=0;
    }
}

void Push(int x){if (!Is_root(x)) Push(Father[x]); Pushdown(x);}
void Splay(int x)
{
    Push(x);
    for (int fa; !Is_root(x); Rotate(x))
        if (!Is_root(fa=Father[x]))
            Rotate(Get(fa)==Get(x)?fa:x);
}

void Access(int x){for (int y=0; x; y=x,x=Father[x]) Splay(x),Son[x][1]=y,Update(x);}
void Make_root(int x){Access(x); Splay(x); Rev[x]^=1;}
int  Find_root(int x){Access(x); Splay(x); while (Son[x][0]) x=Son[x][0]; return x;}
void Link(int x,int y){Make_root(x); Father[x]=y;}

int main()
{
    scanf("%d",&n);
    for (int i=1; i<=n; ++i)
        scanf("%d",&Val[i]);
    scanf("%d",&m);
    for (int i=1; i<=m; ++i)
    {
        scanf("%s%d%d",opt,&x,&y);
        switch (opt[0])
        {
            case 'b':
            {
                if (Find_root(x)==Find_root(y))
                    printf("no\n");
                else
                    printf("yes\n"),Link(x,y);
                break;
            }
            case 'p':
            {
                Access(x); Splay(x);
                Val[x]=y; Update(x);
                break;
            }
            case 'e':
            {
                if (Find_root(x)!=Find_root(y))
                {
                    printf("impossible\n");
                    break;
                }
                Make_root(x);
                Access(y); Splay(y);
                printf("%d\n",Sum[y]);
                break;
            }
        }
    }
}