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

HDU2852

--------明天又要上课了，放假啥也没学貌似，竟玩了。

          题意：有三种操作，0加数， 1删除数， 2找比这个数大的第k个数。

         思路：树状数组的话可以把0~100000范围所有的数做个标记，如果插入了这个数就标记为1，难的地方是2操作

找第k个比a大的数，那就是直接看a这个数后面第k个为1的数，直接从a到100000二分查找就好。

本题代码

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <map>
#include <iostream>
#include <algorithm>
using namespace std;

int c[100005];

int lowbit(int k)
{
    return k & (-k);
}

int sum(int x)
{
    int ans = 0;
    while (x > 0)
    {
        ans += c[x];
        x -= lowbit(x);
    }
    return ans;
}

void add(int x, int val)
{
    while (x <= 100000)
    {
        c[x] += val;
        x += lowbit(x);
    }
}

void PRINT(int a, int k)
{
    int l = a, r = 100000, tmp;
    int now = sum(a);
    while (l < r)
    {
        int mid = (l+r)/2;
        tmp = sum(mid);
        if (tmp < now + k) l = mid + 1;
        else r = mid;
    }
    if (sum(l) >= now + k) printf("%d\n", l);
    else printf("Not Find!\n");
}
int main()
{
    int n;
    while (scanf("%d", &n) == 1)
    {
        memset(c, 0, sizeof c);
        int ope, aa, kk;
        while (n--)
        {
            scanf("%d%d", &ope, &aa);
            if (ope == 0)
            {
                add(aa, 1);
            }
            else if (ope == 1)
            {
                if (sum(aa) - sum(aa-1) == 0) printf("No Elment!\n");
                else add(aa, -1);
            }
            else
            {
                scanf("%d", &kk);
                PRINT(aa, kk);
            }
        }
    }
    return 0;
}