程序设计思维Week5-作业
D-滑动窗口
Description
有一个长度为 n 的数列和一个大小为 k 的窗口, 窗口可以在数列上来回移动. 在窗口从左往右滑的时候,每次窗口内数的最大值和最小值分别是多少?
输入第一行两个整数n和k分别表示数列的长度和滑动窗口的大小,1<=k<=n<=1000000。第二行有n个整数表示ZJM的数列。
输出第一行输出滑动窗口在从左到右的每个位置时,滑动窗口中的最小值。第二行是最大值。
Sample
例如:
数列是 [1 3 -1 -3 5 3 6 7], 其中 k 等于 3.
input:
8 3
1 3 -1 -3 5 3 6 7
output:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
Idea
采用单调队列的思想,与单调栈的操作类似,不同在于队列需要维护左右边界。我们同样用数组q代替单调队列,维护其首尾的索引left、right。
首先我们确定所有大小为k的窗口的最小值,使用一个递增队列,从第一个元素开始,加入队列,当欲加入的元素小于队尾元素时,将队尾元素拿出直至欲加入的元素不小于队尾后加入队列,一开始窗口大小还未达到k,不能马上确定窗口的最小值,当i-k+1大于等于0时表明此时的窗口大小达到了k,可以开始确定每个窗口的最小值即当前队列队首的元素(因为队列是递增的)。此外,我们还需要保证这个窗口在向右移动的过程中大小始终是k,即队列队首元素如果不在当前窗口中它必不是该窗口的最小值,应该及时将队首元素拿出。循环结束后,我们就确定了所有窗口的最小值。
用同样的办法,采用递减队列,可以获得所有窗口的最大值。
Summary
单调队列与单调栈类似,不同之处在于:
①单调队列可以维护首尾两端
②单调队列通常维护局部的单调性
③单调队列通常有大小限制,这里的限制就是k。
因此我们要及时判断队首元素还在不在窗口中。
递增队列求最小值,递减队列求最大值。
同样我们注意到n和k的范围,需要用long long存储它们。
Codes
#include <iostream>
using namespace std;
long long n, k;
int a[1000010],q[1000010],L[1000010],R[1000010];
int main()
{
cin.sync_with_stdio(false);
cin >> n >> k;
for (int i = 0; i < n; i++)
cin >> a[i];
//最小值
int left = 0, right =-1; //递增队列的左右边界
for (int i = 0; i <= n; i++) {
while (right >= left && a[i] < a[q[right]]) right--; //遇到更小的元素
q[++right] = i;
if (q[right] - q[left] + 1 > k)left++; //保证队列中最多只有k个元素
if (i - k + 1 >= 0)L[i - k + 1] = a[q[left]]; //最小值在left上
}
//最大值
left = 0; right = -1; //递减队列的左右边界
for (int i = 0; i <= n; i++) {
while (right >= left && a[i] > a[q[right]]) right--; //遇到更大的元素
q[++right] = i;
if (q[right] - q[left] + 1 > k)left++; //保证队列中最多只有k个元素
if (i - k + 1 >= 0)R[i - k + 1] = a[q[left]]; //最大值在left上
}
for (int i = 0; i <= n - k; i++)
printf("%d ", L[i]);
cout << endl;
for (int i = 0; i <= n - k; i++)
printf("%d ", R[i]);
cout << endl;
}