Week5--作业 -- D -- 滑动窗口[单调队列]

题目描述

ZJM 有一个长度为 n 的数列和一个大小为 k 的窗口, 窗口可以在数列上来回移动. 现在 ZJM 想知道在窗口从左往右滑的时候，每次窗口内数的最大值和最小值分别是多少. 例如：
数列是 [1 3 -1 -3 5 3 6 7], 其中 k 等于 3.

输入

输入有两行。第一行两个整数n和k分别表示数列的长度和滑动窗口的大小，1<=k<=n<=1000000。第二行有n个整数表示ZJM的数列。

输出

输出有两行。第一行输出滑动窗口在从左到右的每个位置时，滑动窗口中的最小值。第二行是最大值。

样例输入

8 3
1 3 -1 -3 5 3 6 7

样例输出

-1 -3 -3 -3 3 3
3 3 5 5 6 7

思路

综述

这道题主要考察了一个新的结构：单调队列

满足以下条件：

1）符合双向队列的一般特性
2）从队首到队尾的元素有序

对于本题

需要用单调队列跑两遍
第一遍找到每个窗口中最小值；
第二遍找到每个窗口中最大值；

每次到一个元素入队的时候需要判断一下条件：

（以队首到队尾升序为例）
1）如果队为空则直接入队；
2）如果队尾元素大于该元素，则不断从队尾出队，直到队列为空或者队尾元素小于该元素，则入队；
3）入队完之后，判断队首元素是否在区间内，不在则队首出队，直到队首的元素在窗口内为止

过程

Step1:初始化

a数组是全局变量，队列中存储的是数组元素的序号

int a[1000050];
	for (int i = 0; i < n; i++) {
		scanf("%d", &a[i]);
	}

Step2:找到每个窗口的最小值

Step2.1：由于第一个区间比较特殊，所以单独处理
代码详细注释：

	for (int i = 0; i < k; i++) {
	//队列为空
		if (qq.empty()) qq.push_back(i);
		//队尾大于该元素的出队
		while (!qq.empty() && a[qq.back()] > a[i]) {
			qq.pop_back();
		}
		//入队
		qq.push_back(i);
	}
	cout << a[qq.front()] << " ";

Step2.2：窗口移动
队首不符合条件的出队

		while (!qq.empty() && qq.front() < l) {
			qq.pop_front();
		}

队尾大于该元素的出队

		while (!qq.empty() && a[qq.back()] > a[i]) {
			qq.pop_back();
		}

安全入队

		qq.push_back(i);

Step3:找到每个窗口的最大值

与找最小值类似，故不赘述

	void getmax() {
	deque<int> qq;
	for (int i = 0; i < k; i++) {
		if (qq.empty()) qq.push_back(i);
		while (!qq.empty() && a[qq.back()] < a[i]) {
			qq.pop_back();
		}
		qq.push_back(i);
	}
	cout << a[qq.front()] << " ";

	int l = 0;
	int r = k - 1;
	for (int i = r + 1; i < n; i++) {
		l++;
		r++;

		while (!qq.empty() && qq.front() < l) {
			qq.pop_front();
		}
		while (!qq.empty() && a[qq.back()] < a[i]) {
			qq.pop_back();
		}
		qq.push_back(i);

		cout << a[qq.front()] << " ";
	}
}

总结

本题和第一题的单调栈类似，STL中的队列，元素一旦出队，就再也找不回了，但是本题采用的是序号入队的方式，巧妙的避免了A题的情况

代码

#include <iostream>
#include <stack>
#include <cstring>
#include <stdio.h>
using namespace std;

int n, k;
int a[1000050];

void getmin() {
	deque<int> qq;
	for (int i = 0; i < k; i++) {
		if (qq.empty()) qq.push_back(i);
		while (!qq.empty() && a[qq.back()] > a[i]) {
			qq.pop_back();
		}
		qq.push_back(i);
	}
	cout << a[qq.front()] << " ";

	int l = 0;
	int r = k - 1;
	for (int i = r + 1; i < n; i++) {
		l++;
		r++;

		while (!qq.empty() && qq.front() < l) {
			qq.pop_front();
		}
		while (!qq.empty() && a[qq.back()] > a[i]) {
			qq.pop_back();
		}
		qq.push_back(i);

		cout << a[qq.front()] << " ";
	}
}
void getmax() {
	deque<int> qq;
	for (int i = 0; i < k; i++) {
		if (qq.empty()) qq.push_back(i);
		while (!qq.empty() && a[qq.back()] < a[i]) {
			qq.pop_back();
		}
		qq.push_back(i);
	}
	cout << a[qq.front()] << " ";

	int l = 0;
	int r = k - 1;
	for (int i = r + 1; i < n; i++) {
		l++;
		r++;

		while (!qq.empty() && qq.front() < l) {
			qq.pop_front();
		}
		while (!qq.empty() && a[qq.back()] < a[i]) {
			qq.pop_back();
		}
		qq.push_back(i);

		cout << a[qq.front()] << " ";
	}
}

int main() {
	cin >> n >> k;
	for (int i = 0; i < n; i++) {
		scanf("%d", &a[i]);
	}
	getmin();
	cout << endl;
	getmax();
}