E - Super Mario HDU - 4417 -主席树第二弹-区间小于K的总数

  • E - Super Mario

  •  HDU - 4417 
  • lower_bound( 1,n,num):从数组的1位置到n位置二分查找第一个大于或等于num的数字.
  • 找到返回该数字的地址,不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。
  • upper_bound( 1,n,num):从数组的1位置到n位置二分查找第一个大于num的数字,
  • 找到返回该数字的地址,不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。
  • 本题需要用upper_bound来查询,返回下标后再-1才是小于等于K的数目。 
  • 题意:长度为n的数组a,m次查询,每次给定一个区间[l,r]和h。求在区间[l,r]内小于等于h的数的个数。
  • 思路:主席树对每一个前缀建一颗线段树,用主席树去维护。线段树中每一个结点代表一个区间。
  • 我们维护结点信息sum,sum表示这个a数组在该区间上的值的个数。这样,求[l,r]中小于等于h的数的个数,
  • 可以通过查询第r个版本的线段树和第l-1个版本的线段树关于“在区间[l,r]内小于等于h的数的个数”的差值来解决。
  • query需要更改,不是找第K大了,是找小于等于K的数目,所以直接用线段树维护的区间跟K比较来确定
  • 这段区间里面的sum有哪些是小于等于K值的sum。
  • #include<bits/stdc++.h>
    using namespace std;
    #define maxn 123456
    int root[maxn],tot,a[maxn],h;
    int n,m,num[maxn],snum,t,l,r;
    struct node
    {
        int l,r,sum;
    } tree[maxn*20];
    int getlow(int x)
    {
        return lower_bound(num+1,num+1+snum,x)-num;
    }
    int getup(int x)
    {
        return upper_bound(num+1,num+1+snum,x)-num-1;
    }
    void build(int &u,int l,int r)
    {
        u=++tot;
        tree[u].sum=0;
        if(l==r)return ;
        int mid=(l+r)/2;
        build(tree[u].l,l,mid);
        build(tree[u].r,mid+1,r);
    }
    void updata(int p,int &u,int l,int r,int k)
    {
        u=++tot;
        tree[u]=tree[p];
        tree[u].sum++;
        if(l==r)return;
        int mid=(l+r)/2;
        if(k<=mid)updata(tree[p].l,tree[u].l,l,mid,k);
        else updata(tree[p].r,tree[u].r,mid+1,r,k);
    }
    int query(int l,int r,int t1,int t2,int k)
    {
        if(l==r)return tree[t2].sum-tree[t1].sum;
        int mid=(l+r)/2;
        if(k<=mid)return query(l,mid,tree[t1].l,tree[t2].l,k);
        else
        {
            int ret=tree[tree[t2].l].sum-tree[tree[t1].l].sum;
            ret+=query(mid+1,r,tree[t1].r,tree[t2].r,k);
            return ret;
        }
    }
    int main()
    {
        scanf("%d",&t);
        for(int cas=1; cas<=t; cas++)
        {
            tot=0;
            scanf("%d%d",&n,&m);
            for(int i=1; i<=n; i++)
            {
                scanf("%d",&a[i]);
                num[i]=a[i];
            }
            sort(num+1,num+1+n);
            snum=unique(num+1,num+1+n)-1-num;
            build(root[0],1,snum);
            for(int i=1; i<=n; i++)
                updata(root[i-1],root[i],1,snum,getlow(a[i]));
            printf("Case %d:\n",cas);
            while(m--)
            {
                scanf("%d%d%d",&l,&r,&h);
                l++,r++;
                int k=getup(h);
                if(k==0)
                    printf("0\n");
                else
                    printf("%d\n",query(1,snum,root[l-1],root[r],k));
            }
        }
        return 0;
    }
    

     

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值