题目描述
南将军统率着N个士兵,士兵分别编号为1~N,南将军经常爱拿某一段编号内杀敌数最高的人与杀敌数最低的人进行比较,计算出两个人的杀敌数差值,用这种方法一方面能鼓舞杀敌数高的人,另一方面也算是批评杀敌数低的人,起到了很好的效果。
所以,南将军经常问军师小工第i号士兵到第j号士兵中,杀敌数最高的人与杀敌数最低的人之间军功差值是多少。
现在,请你写一个程序,帮小工回答南将军每次的询问吧。
注意,南将军可能询问很多次。
输入
只有一组测试数据
第一行是两个整数N,Q,其中N表示士兵的总数。Q表示南将军询问的次数。(1<N<=100000,1<Q<=1000000)
随后的一行有N个整数Vi(0<=Vi<100000000),分别表示每个人的杀敌数。
再之后的Q行,每行有两个正正数m,n,表示南将军询问的是第m号士兵到第n号士兵。
输出
对于每次询问,输出第m号士兵到第n号士兵之间所有士兵杀敌数的最大值与最小值的差。
样例输入
5 2 1 2 6 9 3 1 2 2 4
样例输出
1 7
/*
朴素的暴力暴力算法会超时,因为是O(n^2)的复杂度。
可以用线段树来实现,时间复杂度是O(N*logN)的。
注意m,n的大小关系不确定。
*/
刚开始我的代码是错误的
#include<stdio.h>
#include<algorithm>
using namespace std;
const int maxn=1000010;
int a[maxn];
int main()
{
int N,M,m,n,i;
scanf("%d %d",&N,&M);
for(i=0;i<N;i++)
scanf("%d",&a[i]);
while(M--)
{
scanf("%d %d",&m,&n);
sort(a+m-1,a+n-m);
printf("%d\n",a[n-1]-a[m-1]);
}
return 0;
}
AC码:
修改后,采用ST算法
/*利用ST算法*/
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int N = 100010;
int maxsum[20][N], minsum[20][N]; //优化1
void RMQ(int num) //预处理->O(nlogn)
{
for(int i = 1; i != 20; ++i)
for(int j = 1; j <= num; ++j)
if(j + (1 << i) - 1 <= num)
{//ST算法维护的区间为[j, j + (1 << i) - 1],所以在最后要-1,
maxsum[i][j] = max(maxsum[i - 1][j], maxsum[i - 1][j + (1 << i >> 1)]); //优化2
minsum[i][j] = min(minsum[i - 1][j], minsum[i - 1][j + (1 << i >> 1)]);
}//转移方程
}
int main()
{
int N, Q;
int m, n;
scanf("%d %d", &N, &Q);
for(int i = 1; i <= N; ++i) //输入信息处理
{
scanf("%d", &maxsum[0][i]);
minsum[0][i] = maxsum[0][i];
}
RMQ(N);//ST算法
while(Q--) //O(1)查询
{//m号到n号
scanf("%d %d", &m, &n);
int k = (int)(log(n - m + 1.0) / log(2.0));
int maxres = max(maxsum[k][m], maxsum[k][n - (1 << k) + 1]);
int minres = min(minsum[k][m], minsum[k][n - (1 << k) + 1]);
printf("%d\n", maxres - minres);
}
return 0;
}
还有一种是,建堆栈树
#include <iostream>
#include<algorithm>
#include <cstdio>
#include <cstring>
#define MAXN 1000100
using namespace std;
int array[MAXN];
int Max, Min; //利用全局变量来获取递归结果
typedef struct info{
int Max, Min;
}info;
info segT[MAXN * 4 + 10];
void build(int node, int begin, int end) {
if(begin == end) {
segT[node].Max =
segT[node].Min = array[begin];
}
else {
build(node * 2, begin, (begin + end) / 2);
build(node * 2 + 1, (begin + end) / 2 + 1, end);
segT[node].Max = max(segT[node * 2].Max, segT[node * 2 + 1].Max);
segT[node].Min = min(segT[node * 2].Min, segT[node * 2 + 1].Min);
}
}
void query(int node, int begin, int end, int l, int r) {
if(begin == l && end == r) {
Max = max(Max,segT[node].Max);
Min = min(Min,segT[node].Min);
}
else {
int mid = (begin + end) / 2;
if(r <= mid) query(node * 2, begin, mid, l, r);
else if(l > mid) query(node * 2 + 1, mid + 1, end, l, r);
else {
query(node * 2, begin, mid, l, mid);
query(node * 2 + 1, mid + 1, end, mid + 1, r);
}
}
}
int main() {
int N, M;
while(~scanf("%d %d", &N, &M)) {
int i;
for(i = 1; i <= N; i++) {
scanf("%d", &array[i]);
}
build(1, 1, N);
int l, r;
while(M--)
{
Max = 0; //一定不能忘了初值的设置
Min = 0x7fffffff;
scanf("%d %d", &l, &r);
query(1, 1, N, l, r);
printf("%d\n", Max - Min);
}
}
return 0;
}