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;
}