2733. [HNOI2012]永无乡【平衡树-splay】

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

tmd手残把rotate写错了……
中间将一个splay的点插入到另一个splay中忘了把插入的点clear……
什么鬼啊一个启发式合并的模板题为什么让我做成这样……
累觉不爱

#include<iostream>
#include<cstring>
#include<cstdio>
#define N (100000+100) 
using namespace std;
int n,m,p;
int Root[N],num[N];
int Val[N],Size[N],Father[N],Son[N][2];

int Get(int x) {return Son[Father[x]][1]==x;}
void Update(int x) {Size[x]=Size[Son[x][0]]+Size[Son[x][1]]+1;}
void Clear(int x) {num[x]=Size[x]=Father[x]=Son[x][0]=Son[x][1]=0;}
void Rotate(int x)
{
    int wh=Get(x);
    int fa=Father[x],fafa=Father[fa];
    if (fafa) Son[fafa][Son[fafa][1]==fa]=x;
    Father[fa]=x; Son[fa][wh]=Son[x][wh^1];
    if (Son[fa][wh]) Father[Son[fa][wh]]=fa; 
    Father[x]=fafa; Son[x][wh^1]=fa;
    Update(fa); Update(x); 
}

void Splay(int x)
{
    for (int fa;fa=Father[x];Rotate(x))
        if (Father[fa])
            Rotate(Get(x)==Get(fa)?fa:x);
    Root[num[x]]=x;
}

void Insert(int x,int y)//insert x to y
{
    int now=Root[num[y]],fa=0;
    while (1)
    {
        fa=now,now=Son[now][Val[x]>Val[now]];
        if (now==0)
        {
            Father[x]=fa;
            Son[fa][Val[x]>Val[fa]]=x;
            Size[x]=1;
            num[x]=num[y];
            Update(fa);
            Splay(x);
            return;
        }
    }
}

void Dfs(int x,int y)
{
    if (Son[x][0]) Dfs(Son[x][0],y);
    if (Son[x][1]) Dfs(Son[x][1],y);
    Clear(x); 
    Insert(x,y);
}

void Merge(int x,int y)
{
    if (Size[x]>Size[y]) swap(x,y);
    Dfs(x,y); 
}

int Findx(int now,int x)
{
    while (1)
        if (x<=Size[Son[now][0]])
            now=Son[now][0];
        else
        {
            x-=Size[Son[now][0]];
            if (x==1)
            {
                Splay(now);
                return now;
            }
            x--;
            now=Son[now][1];
        }
}

int main()
{
    char opt[5];
    int x,y;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i)
    {
        scanf("%d",&Val[i]);
        Root[i]=num[i]=i;
        Size[i]=1; 
    }
    //num表示每个点属于的树的标号，Root表示对于标号树的根 
    for (int i=1;i<=m;++i)
    {
        scanf("%d%d",&x,&y);
        if (num[x]!=num[y])
            Merge(Root[num[x]],Root[num[y]]);
    }
    scanf("%d",&p);
    for (int i=1;i<=p;++i) 
    {
        scanf("%s%d%d",opt,&x,&y);
        if (opt[0]=='Q') printf("%d
",Size[Root[num[x]]]>=y?Findx(Root[num[x]],y):-1);
        if (opt[0]=='B' && num[x]!=num[y]) Merge(Root[num[x]],Root[num[y]]);
     }
}