题意:求某个区间内,小于等于h的数有几个
思路:照询问的h从小到大排序,每次把小于等于h的原序列中的数插入(插入的数一定比下一个h小),查询区间内有几个数即可。树状数组、线段树都能做。
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<stdlib.h>
#include<math.h>
#include<vector>
#include<list>
#include<map>
#include<stack>
#include<queue>
#include<algorithm>
#include<numeric>
#include<functional>
using namespace std;
typedef long long ll;
const int maxn = 1e6+5;
struct data
{
int x,cnt;
}node[100005];
struct da
{
int x,y,h,cnt;
}q[100005];
int ans[100005];
int n,tree[maxn];
int cmd(const data &a, const data &b)
{
return a.x < b.x;
}
int cmp(const da &a, const da &b)
{
return a.h < b.h;
}
void init()
{
memset(tree,0,sizeof tree);
}
int lowbit(int k)
{
return k & -k;
}
void add(int cnt,int val)
{
while(cnt <= n) // n -> num of point
{
tree[cnt] += val;
cnt += lowbit(cnt);
}
}
int fid(int cnt) // 1 ~ k sum
{
int sum = 0;
while(cnt)
{
sum += tree[cnt];
cnt -= lowbit(cnt);
}
return sum;
}
int main(void)
{
int T,m,kase = 1;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
init();
for(int i = 1; i <= n; i++)
{
scanf("%d",&node[i].x);
node[i].cnt = i;
}
sort(node+1,node+1+n,cmd);
for(int i = 1; i <= m; i++)
{
scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].h);
q[i].x++;q[i].y++;
q[i].cnt = i;
}
sort(q+1,q+1+m,cmp);
int st = 1;
for(int i = 1; i <= m; i++)
{
while(st <= n && node[st].x <= q[i].h)
{
add(node[st++].cnt,1);
}
int tp = fid(q[i].y) - fid(q[i].x-1);
ans[q[i].cnt] = tp;
}
printf("Case %d:\n",kase++);
for(int i = 1; i <= m; i++)
printf("%d\n",ans[i]);
}
return 0;
}