HDU4417 Super Mario(主席树)

​​link​​

题意：

求区间里$<=h$的数的个数

思路：

• 排序后二分查找小于等于$h$的最大位置pos
• 问题转化为区间$[l,r]$里有多少个数在$[1,pos]$之间。
• 主席树维护某个数出现的次数
• 注意多组输入要清空数组
• 要离散化

代码：

const int maxn=1e5+7,mod=1e9+7,inf=0x3f3f3f3f;

int n,m,a[maxn],b[maxn];
int root[maxn],idx;
struct node{
  int l,r,cnt;
}tr[maxn*40];

int build(int l,int r){
  int p=++idx;
  tr[p].l=tr[p].r=tr[p].cnt=0;
  if(l==r) return p;
  int mid=(l+r)/2;
  tr[p].l=build(l,mid);
  tr[p].r=build(mid+1,r);
  return p;
}

int insert(int p,int l,int r,int x){
  int q=++idx;
  tr[q]=tr[p];
  if(l==r){
    tr[q].cnt++;return q;
  } 
  int mid=(l+r)/2;
  if(x<=mid) tr[q].l=insert(tr[p].l,l,mid,x);
  else tr[q].r=insert(tr[p].r,mid+1,r,x);
  tr[q].cnt=tr[tr[q].l].cnt+tr[tr[q].r].cnt;
  return q;
}

int query(int s,int e,int ql,int qr,int l,int r){
  if(ql<=l&&r<=qr) return tr[e].cnt-tr[s].cnt;
  int mid=(l+r)/2,res=0;
  if(ql<=mid) res+=query(tr[s].l,tr[e].l,ql,qr,l,mid);
  if(qr>mid) res+=query(tr[s].r,tr[e].r,ql,qr,mid+1,r);
  return res;
}


int main(){
  int _=read,Case=0;
  while(_--){
    n=read,m=read;
    rep(i,1,n) a[i]=read,b[i]=a[i];
    sort(b+1,b+1+n);
    int siz=unique(b+1,b+1+n)-(b+1);
    rep(i,1,n) a[i]=lower_bound(b+1,b+1+siz,a[i])-(b);
    idx=0;
    root[0]=build(1,siz);
    rep(i,1,n) 
      root[i]=insert(root[i-1],1,siz,a[i]);
    printf("Case %d:\n",++Case);
    while(m--){
      int l=read+1,r=read+1,h=read;
      int pos=upper_bound(b+1,b+1+siz,h)-b;
      pos--;
      if(!pos) puts("0");
      else{
        printf("%d\n",query(root[l-1],root[r],1,pos,1,siz));
      }
      
    }
  }
  
  
  
  
  
  
  
  
  
  
  
  return 0;
}