题意是给你n个数,然后m组询问,每组为l,r,h,问你在区间[l,r]里有多少个比h小
思路:将n个数从小到大排列,将询问按h从小到大排列,然后按顺序将n个数插入线段树中。具体来说是对于每一个h(从小到大),不停的插入新元素直到当前元素比h大为止,那么可以保证目前线段树上点的值都不大于h,然后就可以区间求和了,复杂度为(nlogn+mlogm)
#include<cstdio>
#include<cstring>
#include<algorithm>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int N=1e5+5;
int sum[N<<2],ans[N],vis[N];
struct node
{
int v,pos;
bool operator <(const node& b)const
{
return v<b.v;
}
}p[N];
struct qq { int l,r,h,id; bool operator <(const qq& b)const { return h<b.h; }
}q[N];
void PushUp(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void update(int pos,int v,int l,int r,int rt)
{
if(l==r)
{
sum[rt]=v;
return;
}
int m=(l+r)>>1;
if(pos<=m)update(pos,v,lson);
else update(pos,v,rson);
PushUp(rt);
}
int query(int L,int R,int l,int r,int rt)
{
if(L<=l&&r<=R)return sum[rt];
int m=(l+r)>>1;
int ans=0;
if(m>=L)ans+=query(L,R,lson);
if(m<R)ans+=query(L,R,rson);
return ans;
}
int main()
{
int T,n,m,kase=0;
//freopen("a.txt","r",stdin);
scanf("%d",&T);
while(T--)
{
printf("Case %d:\n",++kase);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&p[i].v);p[i].pos=i;
}
sort(p+1,p+n+1);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].h);
q[i].l++;q[i].r++;
q[i].id=i;
}
sort(q+1,q+m+1);
memset(sum,0,sizeof(sum));
int k=1;
for(int i=1;i<=m;i++)
{
while(k<=n&&p[k].v<=q[i].h)
{
update(p[k].pos,1,1,n,1);
k++;
}
ans[q[i].id]=query(q[i].l,q[i].r,1,n,1);
}
for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
}
return 0;
}