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

本文介绍了一道经典的树状数组应用题目，通过维护一种特殊的数据结构实现元素的增删及查询大于某值的第k个元素。使用树状数组进行高效更新与查找，结合二分搜索技巧解决了复杂查询需求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

题意：维护一个数据结构，有三种操作：0 e代表把e加入到数据结构中，1 e代表把e从数据结构中删除，2 a k代表求数据结构中大于a的第k大数

思路：此题用树状数组可以轻松过，前两种操作也就是直接更新树状数组，第三种操作二分枚举，当树状数组求和sum(mid) - sum(a) = k的时候，mid就是答案

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int N = 100010;
int n = 100000;
int bit[N], num[N];
int cnt;
void add(int i, int x)
{
    while(i <= n) bit[i] += x, i += i & -i;
}
int sum(int i)
{
    int s = 0;
    while(i > 0) s += bit[i], i -= i & -i;
    return s;
}
int main()
{
    int m, a, b, c;
    while(~ scanf("%d", &m))
    {
        memset(bit, 0, sizeof bit);
        memset(num, 0, sizeof num);
        cnt = 0;
        for(int i = 1; i <= m; i++)
        {
            scanf("%d%d", &a, &b);
            if(a == 0)
            {
                add(b, 1);
                num[b]++;
                cnt++;
            }
            else if(a == 1)
            {
                if(num[b] == 0) printf("No Elment!\n");
                else add(b, -1), cnt--, num[b]--;
            }
            else
            {
                scanf("%d", &c);
                int res = sum(b);
                if(res + c > cnt) printf("Not Find!\n");
                else
                {
                    int l = b, r = n, ans;
                    while(l <= r)
                    {
                        int mid = (l + r) >> 1;
                        if(sum(mid) - res >= c) r = mid - 1, ans = mid;
                        else l = mid + 1;
                    }
                    printf("%d\n", ans);
                }
            }
        }
    }
    return 0;
}