代码源每日一题Div1 108数数

题目链接

思路

我们用树状数组维护区间和，将数组 $a$ 带着下标，按照值大小从小到大排序，再将每个询问，按照 $h$ 值从小到达排序。然后我们遍历每个询问，对于当前的 $h$ ，将数组 $a$ 中小于等于的 $h$的数的下标 $pos$ 进行，$add(pos,1)$。这个询问的答案就是 $ask(r_i)-ask(l_i-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;
}