BZOJ2733永无乡——平衡树启发式合并

Description

永无乡包含 n 座岛，编号从 1 到 n，每座岛都有自己的独一无二的重要度，按照重要度可 以将这 n 座岛排名，名次用 1 到 n 来表示。某些岛之间由巨大的桥连接，通过桥可以从一个岛 到达另一个岛。如果从岛 a 出发经过若干座（含 0 座）桥可以到达岛 b，则称岛 a 和岛 b 是连 通的。现在有两种操作：B x y 表示在岛 x 与岛 y 之间修建一座新桥。Q x k 表示询问当前与岛 x连通的所有岛中第 k 重要的是哪座岛，即所有与岛 x 连通的岛中重要度排名第 k 小的岛是哪 座，请你输出那个岛的编号。

Input

输入文件第一行是用空格隔开的两个正整数 n 和 m，分别 表示岛的个数以及一开始存在的桥数。接下来的一行是用空格隔开的 n 个数，依次描述从岛 1 到岛 n 的重要度排名。随后的 m 行每行是用空格隔开的两个正整数 ai 和 bi，表示一开始就存 在一座连接岛 ai 和岛 bi 的桥。后面剩下的部分描述操作，该部分的第一行是一个正整数 q， 表示一共有 q 个操作，接下来的 q 行依次描述每个操作，操作的格式如上所述，以大写字母 Q 或B 开始，后面跟两个不超过 n 的正整数，字母与数字以及两个数字之间用空格隔开。 对于 20%的数据 n≤1000,q≤1000

对于 100%的数据 n≤100000,m≤n，q≤300000

Output

对于每个 Q x k 操作都要依次输出一行，其中包含一个整数，表 示所询问岛屿的编号。如果该岛屿不存在，则输出-1。

Sample Input
5 1

4 3 2 5 1

1 2

7

Q 3 2

Q 2 1

B 2 3

B 1 5

Q 2 1

Q 2 4

Q 2 3

Sample Output
-1

2

5

1

2

---

平衡树的启发式合并我在平衡树学习(2)——Splay这篇文章里面已经提到过，但是提得比较简单，也就大致讲了一个思路。今天就碰到了一道有关这个的题。

为什么选择splay？因为splay常数大，可以挑战自我。这是因为splay在进行一些操作的时候会比较方便，比如树分裂为小于k和大于k的两棵，那么直接将k所在的位置splay到根，然后直接将左右儿子拆开即可。

而这里的合并是怎么实现的呢?其实很暴力，我们找到要合并的两个根，然后选出其中子树siz较小的哪一个，将它之中的所有点直接插到大的那棵树里即可。是不是很暴力？但复杂度就是能过。总复杂度：$O(nlo{g}^{2}n)$ //代码里merge是合并

然后在维护每个点的信息的时候我们就用并查集记录即可。

#include<bits/stdc++.h>
#define MAXN 100005
using namespace std;
int read(){
    char c;int x;while(c=getchar(),c<'0'||c>'9');x=c-'0';
    while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';return x;
}
void print(int x){
    if(x/10) print(x/10);
    putchar(x%10+'0');
}
int n,m,k,id[MAXN],F[MAXN],root[MAXN];
int fa[MAXN],son[MAXN][2],val[MAXN],siz[MAXN],q[MAXN];
char s;
inline int find(int x){
    if(F[x]!=x)F[x]=find(F[x]);
    return F[x];
}
inline int get(int k){
    return son[fa[k]][1]==k;
}
inline void up(int k){
    siz[k]=siz[son[k][0]]+siz[son[k][1]]+1;
}
inline void rotate(int k){
    int f=fa[k],g=fa[f],d=get(k),p=get(f);
    son[f][d]=son[k][d^1];fa[son[k][d^1]]=f;
    son[k][d^1]=f;fa[f]=k;fa[k]=g;
    if(g) son[g][p]=k;up(f);up(k);
}
inline void splay(int k,int p){
    for(int f;f=fa[k];rotate(k))
      if(fa[f]) rotate(get(k)==get(f)?f:k);
    root[p]=k;
}
inline void insert(int &k,int f,int x,int p){
    if(!k){
        k=p;son[k][0]=son[k][1]=0;
        siz[k]=1;val[k]=x;fa[k]=f;
        return;
    }
    insert(son[k][x>val[k]],k,x,p);
}
inline int query(int k,int x){
    if(!k) return 0;
    if(x<=siz[son[k][0]]) return query(son[k][0],x);
    x-=siz[son[k][0]];
    if(x==1) return val[k];x-=1;
    return query(son[k][1],x);
}
inline void merge(int x,int y){
    if(siz[root[x]]>siz[root[y]])swap(x,y);
    F[x]=y;int h=0,t=0;q[++t]=root[x];
    while(h<t){
        int u=q[++h];
        if(son[u][0]) q[++t]=son[u][0];
        if(son[u][1]) q[++t]=son[u][1];
        insert(root[y],0,val[u],u);
        splay(u,y);
    }
}
int main()
{
    n=read();m=read();id[0]=-1;
    for(int i=1;i<=n;i++){
        val[i]=read();id[val[i]]=i;F[i]=i;root[i]=i;siz[i]=1;
    }
    for(int i=1;i<=m;i++){
        int x=read(),y=read();x=find(x);y=find(y);
        if(x!=y)merge(x,y);
    }
    k=read();
    for(int i=1;i<=k;i++){
        while(s=getchar(),s<'A'||s>'Z');int x=read(),y=read();
        if(s=='B'){
            x=find(x);y=find(y);
            if(x!=y)merge(x,y);
        }
        else{
            x=find(x);
            if(siz[root[find(x)]]<y){puts("-1");continue;}
            print(id[query(root[x],y)]);puts("");
        }
    }
    return 0;
}

吐槽下BZOJ

前两次T是因为写cin，改成scanf直接变成一千多毫秒（这奔腾老爷机真该换了），然后WA是因为写scanf。scanf读空格出现了一些问题，所以就有了我上面那个程序的写法。while(s=getchar(),s<'A'||s>'Z');