HDU 4417 Super Mario 主席树查询区间小于某个值的个数

   

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#include<vector>
#define LL long long
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define per(i,j,k) for(int i=j;i>=k;i--)
#define pb push_back
#define pii pair<int,int>
#define mp make_pair
using namespace std;
const int maxx = 1e5+6;
vector<int>v;
struct node{
  int l,r,w;
}tree[maxx*20];
int cnt;
int root[maxx],a[maxx];
int l[maxx],r[maxx],h[maxx];
int getid(int x){
   return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}
void inserts(int l,int r,int pre,int &now,int pos,int val){
    ///新建节点编号
    now=++cnt;
    ///拷贝并更新节点信息
    tree[now]=tree[pre];
    if (l==r){
            tree[now].w+=val;
        return ;
    }
    int mid=(l+r)>>1;
    if (pos<=mid){
          inserts(l,mid,tree[pre].l,tree[now].l,pos,val);
    }else {
          inserts(mid+1,r,tree[pre].r,tree[now].r,pos,val);
    }
    //维护区间数字的个数目
    tree[now].w=tree[tree[now].l].w+tree[tree[now].r].w;
}
int query(int pre,int now,int l,int r,int pos){
     if (l==r){
        //查询某个区间数字的个数,就是两个版本权值线段树个数的差值
        return tree[now].w-tree[pre].w;
     }
     int mid=(l+r)>>1;
     if (pos<=mid){
      return query(tree[pre].l,tree[now].l,l,mid,pos);
     }else {
      return tree[tree[now].l].w-tree[tree[pre].l].w+query(tree[pre].r,tree[now].r,mid+1,r,pos);
     }
}
int main(){
  int t;
  int n,m;
  scanf("%d",&t);
  int ca=0;
  while(t--){
     ca++;
     printf("Case %d:\n",ca);
     cnt=0;
     scanf("%d%d",&n,&m);
     v.clear();
     memset(root,0,sizeof(root));
     memset(tree,0,sizeof(tree));
     for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        v.pb(a[i]);
     }
     for (int i=1;i<=n;i++){
         scanf("%d%d%d",&l[i],&r[i],&h[i]);
         v.push_back(h[i]);///注意为了防止出错,你需要同时离散高度,防止高度出现问题
     }
     sort(v.begin(),v.end());
     v.erase(unique(v.begin(),v.end()),v.end());
     int sz=v.size();
     for (int i=1;i<=n;i++){
        int x=getid(a[i]);
        inserts(1,sz,root[i-1],root[i],x,1);
     }
     for (int i=1;i<=m;i++){
        int posc=getid(h[i]);
        l[i]++;
        r[i]++;
        if (posc==0)printf("0\n");
        else
        printf("%d\n",query(root[l[i]-1],root[r[i]],1,sz,posc));
     }
  }
  return 0;
}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值