POJ 2985 The k-th Largest Group [并差集+treap]

题意：给出N个数，每个数各自成一个集合，然后M个操作，有两种，0，会将两个数所属的集合并起，1，问当前所有集合中第K大的size是多少

范围: n,m<=20W

解法：很容易联想到并查集，并且当前第K大可以用treap 求，一开始直接塞N个1进去，然后果然T了=  =，于是把等于1的用一个计数器统计一下，就过了，虽然有卡过的嫌疑，但不想改treap了（可以在节点里加一个计数器cot更快的维护，但仔细想想好像就1重复的最多...）。

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<iostream>
#include<stdlib.h>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<bitset>
#pragma comment(linker, "/STACK:1024000000,1024000000")
template <class T>
bool scanff(T &ret){ //Faster Input
    char c; int sgn; T bit=0.1;
    if(c=getchar(),c==EOF) return 0;
    while(c!='-'&&c!='.'&&(c<'0'||c>'9')) c=getchar();
    sgn=(c=='-')?-1:1;
    ret=(c=='-')?0:(c-'0');
    while(c=getchar(),c>='0'&&c<='9') ret=ret*10+(c-'0');
    if(c==' '||c=='\n'){ ret*=sgn; return 1; }
    while(c=getchar(),c>='0'&&c<='9') ret+=(c-'0')*bit,bit/=10;
    ret*=sgn;
    return 1;
}
#define inf 1073741823
#define llinf 4611686018427387903LL
#define PI acos(-1.0)
#define lth (th<<1)
#define rth (th<<1|1)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define drep(i,a,b) for(int i=a;i>=b;i--)
#define gson(i,root) for(int i=ptx[root];~i;i=ed[i].next)
#define tdata int testnum;scanff(testnum);for(int cas=1;cas<=testnum;cas++)
#define mem(x,val) memset(x,val,sizeof(x))
#define mkp(a,b) make_pair(a,b)
#define findx(x) lower_bound(b+1,b+1+bn,x)-b
#define pb(x) push_back(x)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;

const int NN=1000100;

//treap
int tot,root;
struct node{
    int size,l,r,rnd,val;
}t[NN];
void maintain(int x){
    if(x<=0)return;
    t[x].size=t[t[x].l].size+t[t[x].r].size+1;
}
void lturn(int &x){
    int tmp=t[x].r;
    t[x].r=t[tmp].l;
    t[tmp].l=x;
    maintain(x);
    x=tmp;
}
void rturn(int &x){
    int tmp=t[x].l;
    t[x].l=t[tmp].r;
    t[tmp].r=x;
    maintain(x);
    x=tmp;
}
void insert(int &x,int val){
    if(x==0){
        x=++tot;
        t[x].size=1;
        t[x].rnd=rand();
        t[x].l=t[x].r=0;
        t[x].val=val;
    }
    else if(val>t[x].val){
        insert(t[x].r,val);
        if(t[t[x].r].rnd<t[x].rnd)lturn(x);
    }
    else{
        insert(t[x].l,val);
        if(t[t[x].l].rnd<t[x].rnd)rturn(x);
    }
    maintain(x);
}
void dele(int &x,int val){
    if(val>t[x].val)dele(t[x].r,val);
    else if(val<t[x].val)dele(t[x].l,val);
    else{
        if(t[x].l==0)x=t[x].r;
        else if(t[x].r==0)x=t[x].l;
        else{
            if(t[t[x].l].rnd<t[t[x].r].rnd){
                rturn(x);
                dele(t[x].r,val);
            }
            else{
                lturn(x);
                dele(t[x].l,val);
            }
        }
    }
    maintain(x);
}
int findk(int x,int val){
    int lsz=t[t[x].l].size;
    int rsz=t[t[x].r].size;
    if(lsz+1==val)return t[x].val;
    if(val<=lsz)return findk(t[x].l,val);
    return findk(t[x].r,val-lsz-1);
}


//union find set
int f[NN],sz[NN],gnum,num1;
int find(int x){
    return (x==f[x])?x:find(f[x]=f[f[x]]);
}
void connec(int x,int y){
    int i=find(x);
    int j=find(y);
    if(i==j)return;
    gnum--;
    if(sz[i]!=1)dele(root,sz[i]);
    else num1--;
    if(sz[j]!=1)dele(root,sz[j]);
    else num1--;
    f[i]=j;
    sz[j]+=sz[i];
    insert(root,sz[j]);
}

//solve
int n,m;
void init(){
    tot=root=0;
    gnum=num1=n;
    rep(i,1,n)f[i]=i,sz[i]=1;
}
void solve(){
    int op,x,y;
    rep(i,1,m){
        scanff(op);
        scanff(x);
        if(!op){
            scanff(y);
            connec(x,y);
        }
        else {
            int st1=gnum-num1+1;
            if(x>=st1)printf("1\n");
            else printf("%d\n",findk(root,gnum-num1-x+1));
        }
    }
}
int main(){
    while(scanf("%d%d",&n,&m)!=EOF){
        init();
        solve();
    }
    return 0;
}