HDU 2852 线段树（查询有序位置）

题意：有三种操作 操作0 增添数字 操作1 删除数字 操作2 a k 查询大于a的第k个数
这道题与 HDU 4006 POJ 2828 类似
基本上都是找有序位置
详细的解释
传送门

这道题的不同 tree.sum 记录的还是当前结点所含有的数字个数
操作0这里就不多说了 操作1 如何判断是否存在 即判断[1,x] and [1,x-1]这两个区间的sum值是否不同
操作2 即先求出小于等于a的个数 num 再查询第num+k个数就可以了
下面就直接给出代码了

#include <iostream>
#include <cstdio>
#include <cstring>
#define maxn 1000100
using namespace std;
struct tree
{
int l,r,sum;
} tree[maxn*4];
int ans;
//int sum[1000000],ans;
void PushUp(int root)
{
tree[root].sum = tree[root<<1].sum+tree[root<<1|1].sum;
}
void Build(int root,int l,int r)
{
tree[root].l = l;
tree[root].r = r;
if(tree[root].l == tree[root].r)
{
tree[root].sum = 0;
return ;
}
int mid = (tree[root].l+tree[root].r)/2;
Build(root<<1,l,mid);
Build(root<<1|1,mid+1,r);
PushUp(root);
}
void update(int root,int pos,int val)
{
if(tree[root].l==tree[root].r)
{
tree[root].sum += val;
return ;
}
int mid=(tree[root].l+tree[root].r)/2;
if(pos <= mid)
update(root<<1,pos,val);
else update(root<<1|1,pos,val);
PushUp(root);
}
int q_sum(int root,int ql,int qr)
{
if(ql>tree[root].r||qr<tree[root].l)return 0;
if(ql<=tree[root].l&&tree[root].r<=qr)
return tree[root].sum;
int mid=(tree[root].l+tree[root].r)/2;
return q_sum(root<<1,ql,qr)+q_sum(root<<1|1,ql,qr);
}
int find_rt(int root,int t)
{
if(tree[root].l == tree[root].r)
return tree[root].l;
if(t <= tree[root<<1].sum)
return find_rt(root<<1,t);
else
return find_rt(root<<1|1,t-tree[root<<1].sum);
}
int main()
{
int n,m,i,j,p,e,k;
while(~scanf("%d",&m))
{
Build(1,1,maxn);
for(i=0; i<m; i++)
{
scanf("%d",&p);
if(p==0)
{
scanf("%d",&e);
update(1,e,1);
}
else if(p==1)
{
scanf("%d",&e);
if(q_sum(1,1,e)==q_sum(1,1,e-1))
printf("No Elment!\n");
else update(1,e,-1);
}
else if(p==2)
{
scanf("%d%d",&e,&k);
int t=q_sum(1,1,e)+k;
//                cout<<t<<endl;
ans = find_rt(1,t);
if(ans==maxn)
printf("Not Find!\n");
else
printf("%d\n",ans);
}
}
}
return 0;
}