BZOJ2733: [HNOI2012]永无乡（洛谷P3224）

平衡树

BZOJ题目传送门
洛谷题目传送门

平衡树启合裸题。

刚开始把所有节点当成一棵树，每次把小的树合并到大的上，时间复杂度为$O(n\log n)$。

判断是否连通的话用并查集就好了。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define F inline
using namespace std;
struct tree{ int to[2],sz,fa,x; }t[N];
int n,m,rt,a[N],fa[N];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
F int _read(){
    int x=0,f=1; char ch=readc();
    while (!isdigit(ch)&&!isupper(ch)&&ch!='-') ch=readc();
    if (ch=='-') f=-1,ch=readc(); else if (isupper(ch)) return ch;
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x*f;
}
F void writec(int x){
    if (x<0) putchar('-'),x=-x;
    if (x>9) writec(x/10); putchar(x%10+48);
}
F void _write(int x){ writec(x),putchar('\n'); }
int find(int x){ return x==fa[x]?x:fa[x]=find(fa[x]); }
F void pshp(int x){ t[x].sz=t[t[x].to[0]].sz+t[t[x].to[1]].sz+1; }
F void rtt(int x){
    int y=t[x].fa,z=t[y].fa,l=(t[y].to[0]==x);
    if (z) t[z].to[t[z].to[1]==y]=x;
    t[x].fa=z,t[y].to[l^1]=t[x].to[l],t[y].fa=x;
    t[t[y].to[l^1]].fa=y,t[x].to[l]=y,pshp(y),pshp(x);
}
F void splay(int x){
    for (int y=t[x].fa,z=t[y].fa;y;rtt(x),y=t[x].fa,z=t[y].fa)
        if (z) rtt((t[z].to[0]==y^t[y].to[0]==x)?x:y);
    rt=x;
}
F void nsrt(int now){//插入
    int x=rt,i=0,v=t[now].x;
    for (;x;i=x,x=(v<t[x].x?t[x].to[0]:t[x].to[1]));
    x=now,t[i].to[v>t[i].x]=x,t[x].fa=i,splay(x);
}
void clr(int x){//清空
    int l=t[x].to[0],r=t[x].to[1];
    t[x].to[0]=t[x].to[1]=t[x].fa=0,t[x].sz=1;
    nsrt(x); if (l) clr(l); if (r) clr(r);
}
F void mrg(int x,int y){//合并
    if (x==y) return; splay(x),splay(y);
    if (t[x].sz>t[y].sz) swap(x,y);
    fa[x]=y,rt=y,clr(x);
}
int srch(int x,int k){
    int p=t[x].to[0];
    if (t[p].sz+1==k) return x;
    if (t[p].sz>=k) return srch(p,k);
    return srch(t[x].to[1],k-t[p].sz-1);
}
int main(){
    n=_read(),m=_read(),rt=0;
    for (int i=1;i<=n;i++)
        t[i].x=a[i]=_read(),fa[i]=i,t[i].sz=1;
    for (int i=1,x,y;i<=m;i++)
        x=find(_read()),y=find(_read()),mrg(x,y);
    for (int q=_read();q;q--){
        int f=_read(),x=find(_read()),y=_read(),z;
        if (f=='B') y=find(y),mrg(x,y);
        else splay(x),_write((z=srch(x,y))?z:-1);
    }
    return 0;
}