Super Mario HDU - 4417 （离散化线段树+离线处理）

题目链接：https://cn.vjudge.net/problem/HDU-4417

题意：每个样例给出n块砖的高度和m次询问，每次询问给出三个数l,r,h，问区间l-r内<=h的砖有多少块。

思路：
离线处理：

1.把所有砖的高度及下标保存下来，按其高度排序

2.把所有询问及下标保存下来（因为最后要按下标输出结果），然后按询问的高度排序

3.对于高度从小到大的每一个询问，将高度<=它的砖块插入线段树（插入的是砖块的下标），之后进行普通的区间查询（因为这时可以保证线段树内已插入的所有值<=它），然后按询问的下标保存结果

4.按下标从小到大输出结果。

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> pa;   //存储砖块的下标和高度
const int N =1e5+10;
int sum[N<<2],ans[N];
pa a[N];
struct node
{
	int l,r,h,pos;
}q[N];
vector<int> b;
void build(int x,int l,int r)
{
	sum[x]=0;
	if(l==r)
		return;
	int mid=l+r>>1;
	build(x<<1,l,mid);
	build(x<<1|1,mid+1,r);
}
void update(int x,int l,int r,int tem)
{
	if(l==r)
	{
		sum[x]++;
		return;
	}
	int mid=l+r>>1;
	if(tem<=mid)
		update(x<<1,l,mid,tem);
	else
		update(x<<1|1,mid+1,r,tem);
	sum[x]++;
}
int query(int x,int L,int R,int l,int r)
{
	if(L<=l && r<=R)
		return sum[x];
	int mid=l+r>>1,ans=0;
	if(L<=mid)
		ans+=query(x<<1,L,R,l,mid);
	if(R>mid)
		ans+=query(x<<1|1,L,R,mid+1,r);
	return ans;
}
bool cmp(node x,node y)
{
	return x.h<y.h;
}
int main()
{
	int n,m,cas=1,t;
	cin>>t;
	while(t--)
	{
		cin>>n>>m;
		b.clear();
		build(1,1,n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i].first);
			a[i].second=i;
		}
		sort(a+1,a+n+1);  //因为pair类型默认按第一个值排序，不需要自定义cmp函数
		for(int i=0;i<m;i++)
		{
			int l,r,h;
			scanf("%d%d%d",&l,&r,&h);
			q[i]={++l,++r,h,i+1};
		}
		sort(q,q+m,cmp);
		printf("Case %d:\n",cas++);
		for(int i=0,j=1;i<m;i++)
		{
			int l=q[i].l,r=q[i].r,x=q[i].pos;
			while(a[j].first<=q[i].h&&j<=n)
			{
				update(1,1,n,a[j].second);	  //将高度<=它的砖块插入线段树
				j++;
			}	
			ans[x]=query(1,l,r,1,n);		  //按下标存储结果
		}
		for(int i=1;i<=m;i++)
			printf("%d\n",ans[i]);
	}
	return 0;
}