题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2852
题目大意:一共存在三种操作,插入一个数字,删除一个数字,查找大于x的第k大数字是几
题目思路:权值线段树维护每个区间内的数字出现过的次数,插入一个数字就是对应的地方+1,删除就是-1,大于x的第k大,首先得到1~x有多少个数字,这样大于x的第k大就相当于第num+k大数字。
以下是代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
const int MAXN = 1e5+5;
int m,p,x,y;
struct node{
int l,r,val;
}a[MAXN<<2];
void build(int rt,int l,int r){
a[rt].l=l,a[rt].r=r,a[rt].val=0;
if(l==r)return ;
int mid=(l+r)>>1;
build(rt<<1,l,mid);
build(rt<<1|1,mid+1,r);
}
void update(int rt,int pos,int val){
if(a[rt].l==a[rt].r&&a[rt].l==pos){
if(a[rt].val+val<0){
printf("No Elment!\n");
return;
}
a[rt].val+=val;
return;
}
int mid=(a[rt].l+a[rt].r)>>1;
if(pos<=mid)update(rt<<1,pos,val);
else update(rt<<1|1,pos,val);
a[rt].val=a[rt<<1].val+a[rt<<1|1].val;
}
int query1(int rt,int l,int r){
if(a[rt].l>=l&&a[rt].r<=r){
return a[rt].val;
}
int mid=(a[rt].l+a[rt].r)>>1;
int ans=0;
if(l<=mid)ans+=query1(rt<<1,l,r);
if(r>mid)ans+=query1(rt<<1|1,l,r);
return ans;
}
int query2(int rt,int x){
if(a[rt].l==a[rt].r)return a[rt].l;
if(x<=a[rt<<1].val)return query2(rt<<1,x);
else return query2(rt<<1|1,x-a[rt<<1].val);
}
int main(){
while(~scanf("%d",&m)){
build(1,1,1e5);
while(m--){
scanf("%d",&p);
if(!p){
scanf("%d",&x);
update(1,x,1);
}
else if(p==1){
scanf("%d",&x);
update(1,x,-1);
}
else{
scanf("%d%d",&x,&y);
int num=query1(1,1,x);
if(a[1].val<num+y){
printf("Not Find!\n");
continue;
}
int ans=query2(1,num+y);
printf("%d\n",ans);
}
}
}
return 0;
}