hdu2852

题目链接

题意：
给出m个询问
(1)在容器中加入e
(2)在容器中删除e
(3)找出容器中大于a的第k个数

题解：
前两个操作可以在树状数组中完成
第三个操作需要找到一个最左边的数数x，满足sum(x)>=sum(a)+k，这个过程在树状数组上二分即可

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
#include<stack>
#include<map>
#include<cmath>
#include<cstdlib>
#include<string>
#include<vector>
#include<unordered_map>
#include<unordered_set>
using namespace std;
#define scd(n) scanf("%d",&n)
#define scdd(a,b) scanf("%d%d",&a,&b)
#define lowbit(x) x&(-x)
typedef long long LL;
typedef pair<int,int>PII;
const int N = 1e5+7;
int a[N];
int tr[N];
int m,k;

void add(int x,int c)
{
    for(int i=x;i<=N;i+=lowbit(i)) tr[i]+=c;
}

LL sum(int x)
{
    LL res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}

void solve()
{
    memset(tr,0,sizeof tr);
    memset(a,0,sizeof a);
   while(m--){
        int p,e;
        scd(p);
        if(p==0){
            scd(e);
            a[e]++;
            add(e,1);
        }else if(p==1){
            scd(e);
            if(!a[e]){
                puts("No Elment!");
            }else{
                a[e]--;
                add(e,-1);
            }
        }else{
            int a,k;
            scdd(a,k);
            LL s=sum(a);
            int l=1,r=N;
            while(l<r){
                int mid=l+r>>1;
                if(sum(mid)>=s+k) r=mid;
                else l=mid+1;
            }
            if(sum(l)>=s+k){
                printf("%d\n",l);
            }else{
                puts("Not Find!");
            }
        }
   }
}

int main()
{
   //int t;
   //scd(t);
   while(cin >> m){
    solve();
   }
    return 0;
}

标签：int,hdu2852,sum,mid,else,scd,include
来源： https://blog.csdn.net/weixin_51619205/article/details/116983516