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

来源：互联网发布：新人网络男主播技巧编辑：程序博客网时间：2024/05/16 14:01

题意：有三种操作操作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 1000100using 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;}

0 0