POJ2985 The k-th Largest Group(平衡树查询第K大)

传送门
平衡树查询第k大的裸题，用并查集来维护一下哪些猫是一组的就可以了。查询第k大要维护size域，保存一个节点下面的字数的大小，更新的时候从下边更新。

#include<cstdio>
#define MAXN 200005
struct SBT {
    int v, sz, ch[2], cnt;
} t[MAXN];
int n, m, cnt, root;
#define Upd(k) {\
    t[k].sz = t[t[k].ch[0]].sz + t[t[k].ch[1]].sz + t[k].cnt;\
}
void rot(int &k, bool f) {
    int p = t[k].ch[f]; t[k].ch[f] = t[p].ch[!f]; t[p].ch[!f] = k;
    Upd(k); Upd(p); k = p;
}
inline void mt(int &k,bool f)
{
    if(!k) return;
    if(t[t[k].ch[f^1]].sz < t[t[t[k].ch[f]].ch[f]].sz) rot(k, f);
    else if(t[t[k].ch[f^1]].sz < t[t[t[k].ch[f]].ch[f^1]].sz) { rot(t[k].ch[f], f^1); rot(k, f); }
    else return;
    mt(t[k].ch[f],f);
    mt(k,f);
}
void Ins(int &k, int x) {
    if(!k) {k = ++ cnt; t[k].sz = t[k].cnt = 1; t[k].v = x; return; }
    ++ t[k].sz;
    if(t[k].v == x) {++ t[k].cnt; return;}
    Ins(t[k].ch[t[k].v < x], x);
    mt(k, t[k].v < x);
}
int Del(int &k, int x) {
    if(!k) return k;
    int tmp;
    if(t[k].v == x) {
        if(t[k].cnt > 1) {-- t[k].cnt; --t[k].sz; return k;}
        else if(!(t[k].ch[0]*t[k].ch[1])) {k = t[k].ch[0]+t[k].ch[1];}
        else { tmp = Del(t[k].ch[0], x+1); t[k].cnt = t[tmp].cnt; t[k].v = t[tmp].v; Upd(k); return k; }
    }
    else if((t[k].v < x && !t[k].ch[1]) || (t[k].v > x && ! t[k].ch[0])) { tmp = k; k = t[k].ch[0]; Upd(k); return tmp; }
    else tmp = Del(t[k].ch[t[k].v < x], x);
    Upd(k); return tmp;
}
int Kth(int k,int x) {              //查询第k大
    if(x<=t[t[k].ch[1]].sz)
        return Kth(t[k].ch[1],x);
    else if(x>t[t[k].ch[1]].sz+t[k].cnt)
        return Kth(t[k].ch[0],x-t[t[k].ch[1]].sz-t[k].cnt);
    else return t[k].v;
}
char c, f;
inline void GET(int &n) {
    n = 0; f = 1;
    do {c = getchar(); if(c == '-') f = -1;} while(c > '9' || c < '0');
    while(c >= '0' && c <= '9') {n=n*10+c-'0';c=getchar();}
    n *= f;
}
/**********************并查集相关********************/
int fa[MAXN], num[MAXN];
int Q(int x) {
    return x == fa[x] ? x : (fa[x] = Q(fa[x]));
}
void Union(int a, int b) {
    int af = Q(a), bf = Q(b);
    if(af == bf) return;
    Del(root, num[af]); Del(root, num[bf]);
    fa[bf] = af; num[af] += num[bf];
    Ins(root, num[af]);
}
/***************************************************/
int main() {
    GET(n); GET(m);
    int t1, t2, t3;
    Ins(root, 0x7f7f7f7f); Ins(root, -0x7f7f7f7f);
    for(int i = 1; i <= n; ++ i) fa[i] = i, num[i] = 1, Ins(root, 1);
    while(m --) {
        GET(t1); GET(t2);
        if(t1) printf("%d\n", Kth(root, t2+1));
        else {
            GET(t3); Union(t2, t3);
        }
    }
}