HDU4417 Super Mario【树状数组】

题意：求某个区间内，小于等于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;
}