SDU程序设计思维Week5-作业 D-滑动窗口

程序设计思维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;
}