poj2985The k-th Largest Group

链接：http://poj.org/problem?id=2985

题意：初始给定n个元素个数为1的集合，然后给定m个操作，有两种操作：C=0时输入a,b表示将a,b所在的集合合并成一个集合，C=1时输入k表示询问当前所有集合中集合大小排序中的第k大的集合的size。

分析：集合合并直接用并查集维护就可以了，找第k大的话用树状数组找第k大就好了。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=200010;
const int MAX=151;
const int mod=100000000;
const int MOD1=100000007;
const int MOD2=100000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000009;
const ll INF=10000000010;
typedef double db;
typedef unsigned long long ull;
int n,f[N],g[N],p[N];
void add(int x,int y) {
    for (;x<=n;x+=x&(-x)) p[x]+=y;
}
int sum(int x) {
    int ret=0;
    for (;x;x-=x&(-x)) ret+=p[x];
    return ret;
}
int ask_k(int k) {
    int i,w=0;
    for (i=20;i>=0;i--)
    if (w+(1<<i)<=n&&p[w+(1<<i)]<k) {
        k-=p[w+(1<<i)];w+=1<<i;
    }
    return w+1;
}
int find_f(int x) {
    return f[x]==x ? x:f[x]=find_f(f[x]);
}
int main()
{
    int a,b,c,k,i,m,tot;
    scanf("%d%d", &n, &m);
    memset(p,0,sizeof(p));
    for (i=1;i<=n;i++) f[i]=i,g[i]=1;
    add(1,n);tot=n;
    while (m--) {
        scanf("%d", &c);
        if (!c) {
            scanf("%d%d", &a, &b);
            int fa=find_f(a),fb=find_f(b);
            if (fa!=fb) {
                tot--;add(g[fa],-1);add(g[fb],-1);
                f[fa]=fb;g[fb]+=g[fa];add(g[fb],1);
            }
        } else {
            scanf("%d", &k);
            printf("%d\n", ask_k(tot-k+1));
        }
    }
    return 0;
}