0
点赞
收藏
分享

微信扫一扫

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


题目: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;
}



举报

相关推荐

0 条评论