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=100005;
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=0;
    while(i){
        res+=c[i];
        i-=lowbit(i);
    }
    return res;
}

//二分搜索整个序列中第k大的数
int binarySearch(int k){
    int l=1,r=maxn-1,mid,tmp;
    while(r-l>1){
        mid=(l+r)>>1;
        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,0,sizeof(vis));
        memset(c,0,sizeof(c));
        n=0;
        for(int i=1;i<=m;i++){
            scanf("%d",&op);
            if(op==0){
                scanf("%d",&e);
                vis[e]++;
                update(e,1);
                n++;
            }
            else if(op==1){
                scanf("%d",&e);
                if(!vis[e])
                    printf("No Elment!\n");
                else{
                    vis[e]--;
                    update(e,-1);
                    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 0;
}