POJ 3264

Posted 2021-05-13 IdiotNe

一道RMQ问题，锻炼使用线段树解决。不过对于线段树的query过程还有待提高理解程度

#include <iostream>
#include <algorithm>
#include <queue>
#include <string>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <stack>
#include <map>
#include <set>
#include <deque>
using namespace std;

const int maxn= 50005;
const int INF= 0x3f3f3f3f;

struct Node
{
	int l, r;
	int n_min, n_max;
}segTree[maxn<<2];
int n_min, n_max;
int a[maxn];

void Build(int i, int l, int r)
{
	if (l> r){
		return;
	}
	segTree[i].l= l;
	segTree[i].r= r;
	if (l== r){
		segTree[i].n_max= segTree[i].n_min= a[l];
		return;
	}
	int mid= (l+r)>>1;
	Build(i<<1, l, mid);
	Build((i<<1)|1, mid+1, r);
	segTree[i].n_min= min(segTree[i<<1].n_min, segTree[(i<<1)|1].n_min);
	segTree[i].n_max= max(segTree[i<<1].n_max, segTree[(i<<1)|1].n_max);
}
void Query(int i, int l, int r)
{
	if (segTree[i].n_min> n_min && segTree[i].n_max< n_max){
		return;
	}
	if (l== segTree[i].l && r== segTree[i].r){
		n_min= min(segTree[i].n_min, n_min);
		n_max= max(segTree[i].n_max, n_max);
		return;
	}
	int mid= (segTree[i].l+segTree[i].r)>>1;
	if (mid>= r){
		Query(i<<1, l, r);
	}
	else if (mid< l){
		Query((i<<1)|1, l, r);
	}
	else{
		Query(i<<1, l, mid);
		Query((i<<1)|1, mid+1, r);
	}
}

int main(int argc, char const *argv[])
{
	int n, q;
	scanf("%d %d", &n, &q);
	for (int i= 1; i<= n; ++i){
		scanf("%d", a+i);
	}
	Build(1, 1, n);
	while (q--){
		int l, r;
		scanf("%d %d", &l, &r);
		n_max= -INF;
		n_min= INF;
		Query(1, l, r);
		printf("%d\\n", n_max-n_min);
	}
	return 0;
}