HDU

题目链接

题意：

给定一个容器，该容器支持的操作有三种：

1、加入一个元素

2、删除一个元素

3、查询 $k th$ 比 $a$ 大的数

思路：

元素的值在1e5之内，可以直接用树状数组记录出现的次数

查询的时候用二分查找第k大的数。

/*
*  Author: windystreet
*  Date  : 2018-08-10 09:05:15
*  Motto : Think twice, code once.
*/
#include<bits/stdc++.h>

using namespace std;

#define X first
#define Y second
#define eps  1e-5
#define gcd __gcd
#define pb push_back
#define PI acos(-1.0)
#define lowbit(x) (x)&(-x)
#define bug printf("!!!!!
");
#define mem(x,y) memset(x,y,sizeof(x))

typedef long long LL;
typedef long double LD;
typedef pair<int,int> pii;
typedef unsigned long long uLL;

const int maxn = 1e5+7;
const int INF  = 1<<30;
const int mod  = 1e9+7;

int tree[maxn];
int n,sum;
void init(){
    for(int i=0;i<maxn;i++)
        tree[i] = 0;
}
void add(int x,int v){
    for(int i=x;i<maxn;i+=lowbit(i))
        tree[i]+=v;
}
int query(int x){
    int res = 0;
    for(int i=x;i;i-=lowbit(i))
        res += tree[i];
    return res;
}
bool check(int x,int y){
    int k = query(x) - sum;
    return k>=y;                               // 若当前值 >=y 则记录答案
}

void solve(){
    int m,op,x,y;
    while(~scanf("%d",&m)){
        init();                                // 多case不忘初始化
        while(m--){
               scanf("%d",&op);
            if(op==0){
                scanf("%d",&x);
                add(x,1);
            }else if(op==1){
                scanf("%d",&x);
                if(query(x)-query(x-1)){
                    add(x,-1);
                }else{
                    puts("No Elment!");        // 若当前值不存在
                }
            }else{
                scanf("%d%d",&x,&y);
                sum = query(x);                // 记录 [1,x] 区间内元素的个数 
                int L = x, R = maxn-1,ans = -1,mid; 
                while(L<=R){
                    mid = (L + R)>>1;
                    if(check(mid,y)) ans = mid, R = mid-1;
                    else L = mid+1;
                }
                if(ans==-1){
                    puts("Not Find!");
                }else{
                    printf("%d
",ans);
                }
            }
        }
    }
    return;
}

int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
//    ios::sync_with_stdio(false);
    int t = 1;
    //scanf("%d",&t);
    while(t--){
    //    printf("Case %d: ",cas++);
        solve();
    }
    return 0;
}