题目描述:
给出N个数字 Q个询问,每个询问有三个参数 L R H 问 L-R区间内有多少个数小于等于H
题目分析:
先把所有的数字离散化,然后搞莫队,对于每个询问添加删除完事之后,在权值线段树上直接询问有多少个点小于等于H就好了
主席树做法更简单
题目链接:
代码:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
const int maxm=1e5+100;
int val[maxm],Ans[maxm],pos[maxm],_hash[maxm];
int n,m,ans=0;
int cnt,tot;
struct node{
int l,r,h,id;
}a[maxm];
inline bool comp(node x,node y){return pos[x.l]==pos[y.l]?x.r<y.r:x.l<y.l;}
struct tree{
int sum[maxm*4];
inline void pushup(int o){sum[o]=sum[(o<<1)]+sum[(o<<1)|1];}
inline void clear(){memset(sum,0,sizeof(sum));}
inline void insert(int o,int l,int r,int ind,int f)
{
if(l>=r)
{
sum[o]+=f;
return;
}
int mid=(l+r)>>1;
ind<=mid?insert((o<<1),l,mid,ind,f):insert((o<<1)|1,mid+1,r,ind,f);
pushup(o);
}
int ask(int o,int l,int r,int ql,int qr)
{
if(l>qr||r<ql) return 0;
if((ql<=l)&&(r<=qr)) return sum[o];
int mid=(l+r)>>1;
return ask((o<<1),l,mid,ql,qr)+ask((o<<1)|1,mid+1,r,ql,qr);
}
}st;
inline void update(int x,int f){st.insert(1,1,tot,x,f);}
inline void work()
{
st.clear();
ans=0;
cnt=0,tot=0;
scanf("%d%d",&n,&m);
int block_len=std::sqrt(n);
for(int i=1;i<=n;i++) scanf("%d",&val[i]),_hash[++cnt]=val[i],pos[i]=(i-1)/block_len+1;;
for(int i=1;i<=m;i++) scanf("%d%d%d",&a[i].l,&a[i].r,&a[i].h),_hash[++cnt]=a[i].h,a[i].id=i;
std::sort(_hash+1,_hash+cnt+1);
tot=std::unique(_hash+1,_hash+cnt+1)-_hash-1;
for(int i=1;i<=n;i++) val[i]=std::lower_bound(_hash+1,_hash+tot+1,val[i])-_hash;
std::sort(a+1,a+m+1,comp);
int l=1,r=0;
for(int i=1;i<=m;i++)
{
a[i].l++,a[i].r++;
int w=std::lower_bound(_hash+1,_hash+tot+1,a[i].h)-_hash;
while(l<a[i].l) update(val[l++],-1);
while(l>a[i].l) update(val[--l],1);
while(r>a[i].r) update(val[r--],-1);
while(r<a[i].r) update(val[++r],1);
Ans[a[i].id]=st.ask(1,1,tot,1,w);
}
for(int i=1;i<=m;i++) printf("%d\n",Ans[i]);
}
int main()
{
int t;
scanf("%d",&t);
for(int i=1;i<=t;i++)
{
printf("Case %d:\n",i);
work();
}
return 0;
}