POJ - 2985 The k-th Largest Group 树状数组 求第k大

题目链接：POJ - 2985 The k-th Largest Group

利用树状数组前缀和特性（能求出某个数前边有多少个数比他小），将求第K大的数转换为求第 (num-k+1)小的数  num为数的个数。

一开始每个组都只有一只猫，所以一共有n组，此时 c[1]=n （假设 c[ ] 为树状数组）,表示只含一只猫的有n组。我们只需要根据原数组 a[ ] （一开始每组只有一只猫所以初始化全为一）来维护这个树状数组就行。

这里还需要用到并查集的知识，来判断是否属于同一组。更新树状数组时也是根据他们的的父亲节点来更新的

参考了：https://www.cnblogs.com/wuyiqi/archive/2011/12/25/2301071.html 大佬的思路

 Log(n)的复杂度

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=200010;
int pre[maxn],c[maxn],a[maxn];
int n,m;
int find(int x)
{
    return x==pre[x]?x:pre[x]=find(pre[x]);
}
int lowbit(int x)  //x二进制末尾 1 所表示的数
{
    return x&(-x);
}
void update(int x,int v)   //更新树状数组    往x的二进制数后加零
{
    for( ;x<=n;x+=lowbit(x))  //从的二次循环开始，每循环一次就在二进制数后加个零
        c[x] += v;
}
int search(int k)  // 查找
{
    int cnt=0,ans=0;
    for(int i=20;i>=0;i--)  //利用二进制的思想，把答案用一个二进制数来表示
    {
        cnt += (1<<i);
        if(cnt>=n || ans+c[cnt]>=k)
        //这里大于等于k的原因是可能会有很多个数都满足ans + c[cnt] >= k，所以找的是最大的满足ans+c[cnt]<k的cnt
            cnt -= (1<<i);
        else
            ans += c[cnt];  //ans用来累加比当前cnt小的总组数
    }
    //求出的cnt是累加和（即小于等于cnt的数的个数）小于k的情况下cnt的最大值，所以cnt+1就是第k大的数
    return cnt+1;
}
int main()
{
    int k,x,y,num;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        pre[i] = i;
        a[i] = 1;
    }
    num = n;
    update(1,n);  //初始状态值为1的数有n个   
    int p;
    for(int i=0;i<m;i++)
    {
        scanf("%d",&p);
        if(p==0)
        {
            scanf("%d%d",&x,&y);
            x=find(x);
            y=find(y);
            if(x==y)
                continue;
            update(a[x],-1);
            update(a[y],-1);
            update(a[x]+a[y],1);
            pre[y] = x;
            a[x] += a[y];
            num--;
        }
        else
        {
            scanf("%d",&k);
            k = num-k+1;  //转换为求第k小
            printf("%d\n",search(k));
        }
    }
    return 0;
}