欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

滑动窗口

程序员文章站 2022-03-11 19:57:52
...

滑动窗口

#include<bits/stdc++.h>
using namespace std;
int s[10000], ans[10000];
deque<int> Q;
int main() {
	int n, k;
	scanf("%d%d", &n, &k);
	for(int i=0; i<n; i++) {
		scanf("%d", &s[i]);
	}
	for(int i=0; i<n; i++) {
		if(!Q.empty() && i - Q.front() > k-1) Q.pop_front();
		while(!Q.empty() && s[i] < Q.back()) Q.pop_back();
		Q.push_back(i);
		ans[i] = s[Q.front()];
	}
	for(int i=k-1; i<n; i++) printf("%d ", ans[i]);
	Q.clear();
	for(int i=0; i<n; i++) {
		if(!Q.empty() && i - Q.front() > k-1) Q.pop_front();
		while(!Q.empty() && s[i] > Q.back()) Q.pop_back();
		Q.push_back(i);
		ans[i] = s[Q.front()];
	}
	cout << '\n';
	for(int i=k-1; i<n; i++) printf("%d ", ans[i]);
	return 0;
} 

滑动窗口
题目意思:找出数组的每个k长度子区间的最大最小值。
按暴力枚举i,对每个[i, i+k]遍历,找最大最小复杂度 O(nk)。

用单调队列复杂度O(n)。
双端队列 deque,可以对前端后端都可以进行插入删除操作。

先把遍历时产生的各个队列列一下(找最大值):

 1. 1
 2. 3
 3. 3 -1
 4. 3 -1 -3
 5. 5
 6. 5 3
 7. 6
 8. 7

我们观察发现我们每次都会把 s[i] 加进队列,队列中比s[i]小的都pop_back()出去了,剩下的形成了一个单调递减的队列,队列的头就是这个区间的最大值。
但还有特殊情况,例如:给出的s数组就是单调递减的,那上面形成的队列将像这样:

 1. 1 
 2. 1 2
 3. 1 2 3
 4. 1 2 3 4
 5. .......

解决方法:把数组的下标加入队列,这样比较队头元素与i的大小是否超过k-1,超过就pop_front();

相关标签: 题解