HDU2852 KiKi's K-Number 线段树单点更新

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2852


题目大意:给定一个容器有3中操作:“0 x”表示插入元素x;“1 x”表示删除元素x(如果x不存在输出”No Elment!”);“2 x k”表示查询比x大的数中,第k个元素的值(同理,不存在该值输出“Not Find!”)。给你m个操作(m小于10^6),分别模拟每种操作。


分析:很明显,单纯的模拟操作肯定会TLE,那么我们可以用线段树sum来纪录每一区间内元素的总个数,然后对于插入和删除操作,我们分别令该元素所在的区间的sum值加减1即可;对于查询x是否存在,我们可以分别查询[1,x]和[1,x-1]所在区间的sum值,如果不相同,说明x是存在的;找出第k个比x大的元素操作,我们可以把这个操作分为两步,先找出容器内比x小的数有多少个t,即表示出x在容器中是第t大的数,然后我们只需找出容器中第t+k大的数即可。


实现代码如下:

#include <cstdio>
using namespace std;
const int maxn=1000005;
struct segment
{
    int l,r;
    int sum;//纪录区间内有多少个数
}tree[maxn<<2];
void build(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].sum=0;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(root<<1,l,mid);
    build(root<<1|1,mid+1,r);
}
void update(int root,int v,int w)
{
    if(tree[root].l==v&&tree[root].r==v)
    {
        tree[root].sum+=w;
        return ;
    }
    int mid=(tree[root].l+tree[root].r)>>1;
    if(v<=mid) update(root<<1,v,w);
    else update(root<<1|1,v,w);
    tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum;
}
int query(int root,int l,int r)
{
    if(l>tree[root].r||r<tree[root].l)
      return 0;
    if(l<=tree[root].l&&r>=tree[root].r)
      return tree[root].sum;
    int mid=(tree[root].l+tree[root].r)>>1;
    int sum1=0,sum2=0;
    if(l<=mid) sum1=query(root<<1,l,r);
    if(r>mid) sum2=query(root<<1|1,l,r);
    return sum1+sum2;
}
int ans;
void solve(int root,int k)
{
    int mid=(tree[root].l+tree[root].r)>>1;
    if(k>tree[root<<1].sum)
    {
        if(tree[root].l==tree[root].r)
        {
            ans=tree[root].r;
            return ;
        }
        solve(root<<1|1,k-tree[root<<1].sum);
    }
    else solve(root<<1,k);
}
int main()
{
    int m,p,v,k;
    while(scanf("%d",&m)!=-1)
    {
        build(1,1,maxn);
        while(m--)
        {
            scanf("%d",&p);
            if(p==0)
            {
                scanf("%d",&v);
                update(1,v,1);
            }
            else if(p==1)
            {
                scanf("%d",&v);
                if(query(1,1,v)==query(1,1,v-1))
                  puts("No Elment!");
                else update(1,v,-1);
            }
            else
            {
                scanf("%d%d",&v,&k);
                int t=query(1,1,v);
                //printf("%d\n",t);
                solve(1,t+k);
                if(ans==maxn) puts("Not Find!");
                else printf("%d\n",ans);
            }
        }
    }
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值