题意:给一个长度n的数列,输出其中每个长度为k的区间的最小值和最大值。
分析:一共需要求(n-k+1)次区间的最大值和最小值,用两次单调队列,一次维护最小值,一次维护最大值,复杂度O(N)。另外线段树也可以,复杂度O(N*logN)。
先维护一个单调递增序列,对于每个元素,如果大于队尾元素(不破坏队列单调性),就直接插到队尾;如果小于队首元素,就不断弹出队尾元素,知道队列为空或队尾元素小于当前元素,此时再把当前元素插到队尾。这样每次读取结果的时候保证了队首元素就是最小元素。此外还要判断队首元素的下标是否超出范围,超出范围就弹出队首,直到下标在范围内。求最大值只需要维护一个单调递减队列。
输入数据较大,加了一个输入挂,摘自 http://www.cnblogs.com/xiaohongmao/archive/2012/05/29/2523760.html
另外要用C++提交,G++会TLE!!!这里坑了好久,待求解。。。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int mod = 1000000007;
const int maxn = 1e6 + 5;
int a[maxn],mi[maxn],ma[maxn],m1[maxn],m2[maxn];
inline bool scan_d(int &num)
{
char in;bool IsN=false;
in=getchar();
if(in==EOF) return false;
while(in!='-'&&(in<'0'||in>'9')) in=getchar();
if(in=='-'){ IsN=true;num=0;}
else num=in-'0';
while(in=getchar(),in>='0'&&in<='9'){
num*=10,num+=in-'0';
}
if(IsN) num=-num;
return true;
}
int main(){
int n,k;
while(~scanf("%d%d", &n, &k)){
for(int i=1; i<=n; ++i){
scan_d(a[i]);
}
int minl = 0, minr = 0, maxl = 0, maxr = 0;
for(int i=1; i<k; ++i){
/*先让前k-1个元素入队*/
while(minr > minl && a[mi[minr - 1]] >= a[i]){
--minr;
}
mi[minr++] = i;
while(maxr > maxl && a[ma[maxr - 1]] <= a[i]){
--maxr;
}
ma[maxr++] = i;
}
for(int i=k; i<=n; ++i){
/*逐个处理后面n-k+1个元素*/
while(minr > minl && a[mi[minr - 1]] >= a[i]){
/*维护单调性*/
--minr;
}
mi[minr++] = i;/*新元素入队*/
while(minl < minr && mi[minl] <= i - k){
/*保证下标在范围内*/
++minl;
}
m1[i - k] = a[mi[minl]];
while(maxr > maxl && a[ma[maxr - 1]] <= a[i]){
--maxr;
}
ma[maxr++] = i;
while(maxl < maxr && ma[maxl] <= i - k){
++maxl;
}
m2[i - k] = a[ma[maxl]];
}
for(int i=0; i<n-k; ++i){
printf("%d ", m1[i]);
}
printf("%d\n", m1[n - k]);
for(int i=0; i<n-k; ++i){
printf("%d ", m2[i]);
}
printf("%d\n", m2[n - k]);
}
return 0;
}