思路
我们用树状数组维护区间和,将数组 a a a 带着下标,按照值大小从小到大排序,再将每个询问,按照 h h h 值从小到达排序。然后我们遍历每个询问,对于当前的 h h h ,将数组 a a a 中小于等于的 h h h的数的下标 p o s pos pos 进行, a d d ( p o s , 1 ) add(pos,1) add(pos,1)。这个询问的答案就是 a s k ( r i ) − a s k ( l i − 1 ) 。 ask(r_i) - ask(l_i-1)。 ask(ri)−ask(li−1)。
代码
#include<bits/stdc++.h>
#define rep(i,x,y) for(int i=x; i<=y; ++i)
#define per(i,x,y) for(int i=x; i>=y; --i)
#define pushk push_back
#define popk pop_back
#define mem(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define ll long long
#define lp p<<1
#define rp p<<1|1
using namespace std;
const int N = 1e5+9;
int tr[N];
int lowbit(int x){
return x&-x;
}
void add(int p,int c){
for(int i=p; i<N; i+=lowbit(i)) tr[i]+=c;
}
int ask(int p){
int res=0;
for(int i=p; i; i-=lowbit(i)) res+=tr[i];
return res;
}
struct node{
int l,r,h,pos;
bool operator<(const node &t) const{
return h<t.h;
}
}q[N];
struct Num{
int num,pos;
bool operator<(const Num &t) const{
return num<t.num;
}
}a[N];
int ans[N];
int main() {
int T;
cin>>T;
while(T--){
int n,Q;
scanf("%d %d",&n,&Q);
mem(tr,0);
rep(i,1,n) scanf("%d",&a[i].num),a[i].pos=i;
sort(a+1,a+n+1);
rep(i,1,Q){
int l,r,h;
scanf("%d %d %d",&l,&r,&h);
q[i]={l,r,h,i};
}
sort(q+1,q+Q+1);
int idx=1;
rep(i,1,Q){
while(a[idx].num<=q[i].h && idx<=n){
add(a[idx].pos,1);
idx++;
}
ans[q[i].pos]=ask(q[i].r) - ask(q[i].l-1);
}
rep(i,1,Q) printf("%d ",ans[i]);
puts("");
}
return 0;
}