HDU 2852 KiKi's K-Number(树状数组+二分搜索)

题意:给出三种操作
  0 e:将e放入容器中
  1 e:将e从容器中删除,若不存在,则输出No Elment!
  2 a k:搜索容器中比a大的第k个数,若不存在,则输出Not Find!

思路:树状数组+二分搜索,具体见代码吧。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
/*
AC
树状数组+二分搜索
题意:给出三种操作
0 e:将e放入容器中
1 e:将e从容器中删除,若不存在,则输出No Elment!
2 a k:搜索容器中比a大的第k个数,若不存在,则输出Not Find! 思路:树状数组+二分搜索,具体见代码吧。
*/
using namespace std;
const int maxn=;
int m;
int c[maxn]; //树状数组
int vis[maxn]; //vis[i]表示元素i的个数 int lowbit(int x){
return x&(-x);
} void update(int i,int v){
while(i<maxn){
c[i]+=v;
i+=lowbit(i);
}
} //sum(i)表示容器中元素大小在1~i之间的个数
int sum(int i){
int res=;
while(i){
res+=c[i];
i-=lowbit(i);
}
return res;
} //二分搜索整个序列中第k大的数
int binarySearch(int k){
int l=,r=maxn-,mid,tmp;
while(r-l>){
mid=(l+r)>>;
tmp=sum(mid);
if(k<=tmp)
r=mid;
else
l=mid;
}
return r;
}
int main()
{
int op,e,k;
int n; //目前存放的元素个数
while(scanf("%d",&m)!=EOF){
memset(vis,,sizeof(vis));
memset(c,,sizeof(c));
n=;
for(int i=;i<=m;i++){
scanf("%d",&op);
if(op==){
scanf("%d",&e);
vis[e]++;
update(e,);
n++;
}
else if(op==){
scanf("%d",&e);
if(!vis[e])
printf("No Elment!\n");
else{
vis[e]--;
update(e,-);
n--;
}
}
else{
scanf("%d%d",&e,&k);
int t=sum(e); // 目前<=e的元素有t个,转化成搜索第t+k个大的元素
//若t+k直接大于n,则不存在该元素
if(t+k>n)
printf("Not Find!\n");
else{
k=k+t;
printf("%d\n",binarySearch(k));
}
}
}
}
return ;
}
上一篇:[JS代码]如何判断ipad或者iphone是否为横屏或者竖屏 - portrait或者landscape


下一篇:django2.0再写一行代码