首先引入一个问题,原网址http://poj.org/problem?id=2823,本人翻译如下:
题目描述
给一个大小为n的数组(n ≤ 106)。有一个大小为k的滑动窗,从阵列的最左边移动到最右边。你只能在窗中看到K个数字。每次滑动窗向右移动一个位置。以下为示例:
窗的位置 | 最小值 | 最大值 |
[1 3 -1] -3 5 3 6 7 | -1 | 3 |
1 [3 -1 -3] 5 3 6 7 | -3 | 3 |
1 3 [-1 -3 5] 3 6 7 | -3 | 5 |
1 3 -1 [-3 5 3] 6 7 | -3 | 5 |
1 3 -1 -3 [5 3 6] 7 | 3 | 6 |
1 3 -1 -3 5 [3 6 7] | 3 | 7 |
数组为[1 3 -1 -35 3 6 7],k为3。
输入
输入由两行组成。第一行包含两个整数n和k,它们分别是数组和滑动窗口的长度。第二行有n个整数。
输出
输出中有两行。第一行从左到右给出窗中每个位置的最小值,第二行给出最大值。
输入样例
8 3
1 3 -1 -3 5 3 6 7
输出样例
-1 -3 -3 -3 3 3
3 3 5 5 6 7
将窗放上去,然后找到这最开始的k个数的最大值,然后窗最后移一个单元,继续找到k个数中的最大值。这是每个人都会的最直观的暴力求解法。然而看看n的数据规模,你的心情一定是这样的
那么怎么办呢?
我们可以引入单调队列思想。单调,顾名思义,就是数学必修一那个:)。显然,单调区间内的最值一定在最左或最右。用这个思路来解决这道题,可以避免很多不必要的重复判断,保存上次的结果。此算法的复杂度和空间都远远优于暴力求解。
单调队列的定义如下:
1、维护区间的最值。
2、去除冗杂状态如上题,区间中的两个元素a[i],a[j]。(假设现在再求最大值)
若 j>i且a[j]>=a[i],a[j]比a[i]还大而且还在后面(目前a[j]留在队列肯定比a[i]有用,因为你是往后推,核心思想!)
3、维护队列的单调性,最大值是单调递减序列,最小值反之。
4、最优选择在队首。
用单调队列解决的程序如下:
#include<bits/stdc++.h>
using namespace std;
struct node
{
int x,y;
}v[1010000]; //x表示值,y表示位置 可以理解为下标
int a[1010000],n,m,mx[1010000],mn[1010000];
void getmin()
{
int i,head=1,tail=0;// 默认起始位置为1 因为插入是v[++tail]故初始化为0
for(i=1;i<m;i++)
{
while(head<=tail && v[tail].x>=a[i]) tail--;//删除不满足单调性的点
v[++tail].x=a[i],v[tail].y=i;
// 根据题目 前m-1个先直接进入队列
}
for(i=m;i<=n;i++)//滑窗口,摩擦摩擦(i为窗内的第一个数)
{
while(head<=tail && v[tail].x>=a[i]) tail--;//删除不满足单调性的点
v[++tail].x=a[i],v[tail].y=i;
while(v[head].y<i-m+1) head++;
mn[i-m+1]=v[head].x;
// 道理同上。当然了,要把已经超出范围的从head开始排出
// 然后每个队首则是目前m个数的最小值
}
}
void getmax() //最大值同最小值的道理,只不过是维护的是递减队列
{
int i,head=1,tail=0;
for(i=1;i<m;i++)
{
while(head<=tail && v[tail].x<=a[i]) tail--;
v[++tail].x=a[i],v[tail].y=i;
}
for(i=m;i<=n;i++)
{
while(head<=tail && v[tail].x<=a[i]) tail--;
v[++tail].x=a[i],v[tail].y=i;
while(v[head].y<i-m+1) head++;
mx[i-m+1]=v[head].x;
}
}
int main()
{
int i,j;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++) scanf("%d",&a[i]);
getmin();
getmax();
for(i=1;i<=n-m+1;i++)
{
if(i==1) printf("%d",mn[i]);
else printf(" %d",mn[i]);
}
printf("\n");
for(i=1;i<=n-m+1;i++)
{
if(i==1) printf("%d",mx[i]);
else printf(" %d",mx[i]);
}
printf("\n");
return 0;
}