POJ3264 Balanced Lineup(USACO 2007 January Silver)

Balanced Lineup

题目链接-Balanced Lineup

Sample Input

6 3
1
7
3
4
2
5
1 5
4 6
2 2

Sample Output

6
3
0

题目大意
农夫有n头牛，给你这n头牛各自的高度，q次询问，问[a,b]区间内最高的牛和最矮的牛身高差异

解题思路
线段树的区间查找最大最小值模板

• query函数中加一个bool类型的ass，用来判断是算区间最大值还是最小值，这样避免了写两个函数分别求最大值和最小值
• 记住输入输出用scanf和printf，不然会超时
• 详细解析见代码

附上代码

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
const int N=5e4+5;
int a[N];
struct node{
	int l,r,minn,maxx;
}tr[N<<2];
void push_up(int i){//更新
	tr[i].minn=min(tr[i<<1].minn,tr[i<<1|1].minn);
	tr[i].maxx=max(tr[i<<1].maxx,tr[i<<1|1].maxx);
}
void build(int l,int r,int i){//建树
	//i为当前需要建立的结点，l为当前需要建立区间的左端点，r则为右端点
	tr[i].l=l;tr[i].r=r;
	if(l==r){//左端点等于右端点，即为叶子节点，直接赋值即可
		tr[i].minn=tr[i].maxx=a[l];
		return ;
	}
	int m=l+((r-l)>>1);
	//m则为中间点，左儿子的结点区间为[l,m],右儿子的结点区间为[m+1,r]
	build(l,m,i<<1);//构造左儿子节点
	build(m+1,r,i<<1|1);//构造右儿子节点
	push_up(i);//更新父节点
}
int query(int l,int r,int i,int ass){//查询
	if(l<=tr[i].l&&tr[i].r<=r){
	//如果当前结点的区间真包含于要查询的区间内，则返回结点信息且不需要往下递归
		return ass?tr[i].maxx:tr[i].minn;
	}
	int res;
	int m=tr[i].l+((tr[i].r-tr[i].l)>>1);
	if(ass){//ass=1时查询最大值
		res=-INF;
		if(l<=m)//如果左子树和需要查询的区间交集非空
			res=max(res,query(l,r,i<<1,ass));
		if(m<r)//如果右子树和需要查询的区间交集非空，不能用else,因为查询区间可能同时和左右区间都有交集
			res=max(res,query(l,r,i<<1|1,ass));
	}
	else{
		res=INF;
		if(l<=m)
			res=min(res,query(l,r,i<<1,ass));
		if(m<r)
			res=min(res,query(l,r,i<<1|1,ass));
	}
	return res;
}
signed main(){
	
	int n,q;
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	build(1,n,1);
	while(q--){
		int a,b;
		scanf("%d%d",&a,&b);
		printf("%d\n",query(a,b,1,1)-query(a,b,1,0));
	}
	return 0;
}