题目链接: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;
}