hdu-2852 KiKi's K-Number---二分+树状数组

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=2852

题目大意：

题意：
    给出三种操作，
    0 在容器中插入一个数。
    1 在容器中删除一个数。
    2 求出容器中大于a的第k大元素。

解题思路：

用树状数组维护每个值，插入数字是add(x, 1),删除时add(x, -1)

查询第k大时，先判断是否存在，存在的话直接根据树状数组sum值的单调性二分法求解即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<map>
#include<set>
#include<cmath>
#include<algorithm>
#include<vector>
#include<sstream>
#define lowbot(i) (i&(-i))
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 10;
int tree[maxn];
void add(int x, int d)
{
    while(x <= 100000)
        tree[x] += d, x += lowbot(x);
}
int sum(int x)
{
    int ans = 0;
    while(x)ans += tree[x], x -= lowbot(x);
    return ans;
}
int Find(int x, int k)
{
    int l = x + 1, r = 100000;
    int mid, t;
    while(l < r)
    {
        //cout<<l<<" "<<r<<endl;
        mid = (l + r) / 2;
        t = sum(mid) - sum(x);
        if(t >= k)
            r = mid;
        else l = mid + 1;
    }
    return r;
}
int main()
{
    int n, x, k;
    while(scanf("%d", &n) != EOF)
    {
        memset(tree, 0, sizeof(tree));
        while(n--)
        {
            scanf("%d", &x);
            if(x == 0)
            {
                scanf("%d", &x);
                add(x, 1);
            }
            else if(x == 1)
            {
                scanf("%d", &x);
                if(sum(x) == sum(x - 1))
                    printf("No Elment!
");
                else add(x, -1);
            }
            else if(x == 2)
            {
                scanf("%d%d", &x, &k);
                if(sum(100000) - sum(x) < k)
                    printf("Not Find!
");
                else
                {
                    printf("%d
", Find(x, k));
                }
            }
        }
    }
    return 0;
}