洛谷 P1886 滑动窗口 /【模板】单调队列 (自己造轮子和STL的deque)

题目描述

有一个长为 n 的序列 a，以及一个大小为 k 的窗口。现在这个从左边开始向右滑动，每次滑动一个单位，求出每次滑动后窗口中的最大值和最小值。

例如，对于序列 [1,3,−1,−3,5,3,6,7] 以及 k=3，有如下过程：

输入格式

输入一共有两行，第一行有两个正整数 n,k。 第二行 n 个整数，表示序列 a

输出格式

输出共两行，第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值

输入输出样例

输入 #1复制

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

输出 #1复制

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

说明/提示

【数据范围】
对于 50% 的数据，1≤n≤10^5；
对于 100% 的数据，1≤k≤n≤10^6，ai∈[−2^31,2^31)。

这道题的题目已经讲的非常清楚了，就是滑动窗口，那本质就是单调队列，维护这个队列的单调性

比如我们看用例的前三个点（以单调递增为例）

很明显前两个点高于第三个点，把前两个点出队，维护单调性，加入第三个点

我们再看接下来的三个点

很明显在第一个点高于第二个点，那么先将第一个点出队维护单调性，加入第二个点，然后第三个点高于第二个点直接加入，以此类推

用数组自己造的队列代码如下（y总说复杂度小于STL哦）：

#include <iostream>

using namespace std;

const int N = 1e6 + 10;
int a[N],q[N];

int n,k;

int main()
{
	cin >> n >> k;
	for(int i = 0;i < n;i ++) cin >> a[i];

	int hh = 0,tt = -1;
	for(int i = 0;i < n;i ++){
		while(hh <= tt && i - k + 1 > q[hh]) hh ++;

		while(hh <= tt && a[q[tt]] > a[i]) tt --;

		q[++ tt] = i;

		if(i >= k - 1) cout << a[q[hh]] << " ";
		// cout << tt << endl;
	}

	cout << endl;

	hh = 0,tt = -1;
	for(int i = 0;i < n;i ++){
		while(hh <= tt && i - k + 1 > q[hh]) hh ++;

		while(hh <= tt && a[q[tt]] < a[i]) tt --;

		q[++ tt] = i;

		if(i >= k - 1) cout << a[q[hh]] << " ";
	}

	return 0; 
}

STL代码如下：

#include <iostream>
#include <deque>

using namespace std;

const int N = 1e6 + 10;
int a[N];

int n, k;

int main() {
    cin >> n >> k;
    for(int i = 0; i < n; i++) cin >> a[i];

    deque<int> q;

    // 计算滑动窗口的最小值
    for(int i = 0; i < n; i++) {
        if(!q.empty() && q.front() < i - k + 1) q.pop_front(); // 移除超出窗口的元素

        while(!q.empty() && a[q.back()] > a[i]) q.pop_back(); // 保持队列单调递增

        q.push_back(i); // 将当前元素的下标加入队列

        if(i >= k - 1) cout << a[q.front()] << " "; // 输出当前窗口的最小值
    }

    cout << endl;
    q.clear(); // 清空队列

    // 计算滑动窗口的最大值
    for(int i = 0; i < n; i++) {
        if(!q.empty() && q.front() < i - k + 1) q.pop_front(); // 移除超出窗口的元素

        while(!q.empty() && a[q.back()] < a[i]) q.pop_back(); // 保持队列单调递减

        q.push_back(i); // 将当前元素的下标加入队列

        if(i >= k - 1) cout << a[q.front()] << " "; // 输出当前窗口的最大值
    }

    return 0;
}

加油