bzoj 2733【hnoi2012】永无乡（平衡树）

时间限制：1秒  内存限制：64M

【问题描述】

　　永无乡包含 n 座岛，编号从 1 到 n，每座岛都有自己的独一无二的重要度，按照重要度可 以将这 n 座岛排名，名次用 1 到 n 来表示。某些岛之间由巨大的桥连接，通过桥可以从一个岛 到达另一个岛。如果从岛 a 出发经过若干座（含 0 座）桥可以到达岛 b，则称岛 a 和岛 b 是连 通的。现在有两种操作：

　　B x y 表示在岛 x 与岛 y 之间修建一座新桥。

　　Q x k 表示询问当前与岛 x连通的所有岛中第 k 重要的是哪座岛，即所有与岛 x 连通的岛中重要度排名第 k 小的岛是哪座，请你输出那个岛的编号。

【输入格式】

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

【输出格式】

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

【输入样例】

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

【输出样例】

-1
2
5
1
2

【数据范围】

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

【来源】

bzoj 2733

这道题就是一道数据结构题，我们用一个伪并查集和一个平衡树就可以解决，至于加点，我们可以证明有意义的点的变化不会超过n*log2(n)，所以直接一个一个的加就可以了。

代码如下：

#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=100005;

int a[maxn],ch[maxn][2],fa[maxn],pa[maxn],root[maxn],n,s[maxn];

int find(int x){
    if(pa[x]==x) return x;
    return pa[x]=find(pa[x]);
}

void up(int x){
    s[x]=s[ch[x][0]]+s[ch[x][1]]+1;
}

void link(int x,int d,int y){
    fa[y]=x;ch[x][d]=y;
}

void rot(int x){
    int y=fa[x],z=fa[y];
    int d=(ch[y][0]==x);
    link(y,d^1,ch[x][d]);
    link(z,ch[z][1]==y,x);
    link(x,d,y);
    up(y);
    up(x);
}

void sp(int x){
    int y,z;
    while(fa[x])
    {
        y=fa[x],z=fa[y];
        if(z)
        if((ch[z][1]==y)==(ch[y][1]==x)) rot(y);else rot(x);
        rot(x);
    }
    int px=find(x);
    root[px]=x;
}

void in(int x,int y){
    int t=y,d;
    while(1)
    {
        if(a[x]<=a[t]) d=0;
        else d=1;
        if(!ch[t][d]) break;
        t=ch[t][d];
    }
    link(t,d,x);
    sp(x);
}

void add(int x,int y){
    if(ch[x][0]) add(ch[x][0],y);
    if(ch[x][1]) add(ch[x][1],y);
    in(x,y);
}

int K(int x,int k){
    int t,p=x;
    while(1)
    {
        t=1;
        if(ch[p][0]) t+=s[ch[p][0]];
        if(t>k) p=ch[p][0];
        else if(t<k) p=ch[p][1],k-=t;
        else break;
    }
    return p;
}

int read(){
    int x=0;
    char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x;
}

void init()
{
    n=read();int m=read();
    for(int i=1;i<=n;i++) 
    {
        a[i]=read();
        root[i]=i;
        pa[i]=i;
        s[i]=1;
    }
    s[0]=0;
    int x,y,px,py;
    while(m--)
    {
        x=read();y=read();
        px=find(x),py=find(y);
        if(px==py) continue;
        if(s[root[px]]>s[root[py]]) swap(px,py);
        pa[px]=py;
        add(root[px],root[py]);
    }
}
int main()
{
    //freopen("B.in","r",stdin);
    //freopen("B.out","w",stdout);
    init();
    int T=read(),x,y,px,py;
    char q[5];
    while(T--)
    {
        scanf("%s",q);
        if(q[0]=='B')
        {
            x=read();y=read();
            px=find(x),py=find(y);
            if(px==py) continue;
            if(s[root[px]]>s[root[py]]) swap(px,py);
            pa[px]=py;
            add(root[px],root[py]);
        }
        if(q[0]=='Q')
        {
            x=read();y=read();
            px=find(x);
            if(s[root[px]]<y) printf("-1\n");
            else printf("%d\n",K(root[px],y));
        }
    }
    return 0;
}