滑动窗口
https://ac.nowcoder.com/acm/problem/50528
https://ac.nowcoder.com/discuss/394776
题目描述
给一个长度为N的数组,一个长为K的滑动窗体从最左端移至最右端,你只能看到窗口中的K个数,每次窗体向右移动一位,如下图:
你的任务是找出窗体在各个位置时的最大值和最小值。
输入描述:
第1行:两个整数N和K;
第2行:N个整数,表示数组的N个元素(
≤
2
×
1
0
9
≤2 \times10^9
≤2×109);
输出描述:
第一行为滑动窗口从左向右移动到每个位置时的最小值,每个数之间用一个空格分开;
第二行为滑动窗口从左向右移动到每个位置时的最大值,每个数之间用一个空格分开。
示例1
输入
8 3
1 3 -1 -3 5 3 6 7
输出
-1 -3 -3 -3 3 3
3 3 5 5 6 7
备注:
对于
20
%
20 \%
20%的数据,
K
≤
N
≤
1000
K≤N≤1000
K≤N≤1000;
对于
50
%
50 \%
50%的数据,
K
≤
N
≤
1
0
5
K≤N≤10^5
K≤N≤105;
对于
100
%
100 \%
100%的数据,
K
≤
N
≤
1
0
6
K≤N≤10^6
K≤N≤106。
思路:
单调队列的入门题。维护一个队列的单调性,如果要求滑动窗口的最大值,那么要保持队首为最大值,也就是单调递减,因为窗口要往后移,不属于此窗口中的元素就要出队,那么最终留在队首的元素就是在此窗口范围内的最大值。
Code:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 5;
int a[maxn];
int q[maxn]; //手写队列
int main()
{
int n, k;
cin >> n >> k;
for (int i = 1; i <= n; i++)
cin >> a[i];
int l = 1, r = 0;
for (int i = 1; i <= n; i++)
{
while (l <= r && a[q[r]] > a[i])
r--;
while (l <= r && q[l] + k <= i)
l++;
q[++r] = i;
if (i >= k)
cout << a[q[l]] << " ";
}
cout << endl;
l = 1, r = 0;
for (int i = 1; i <= n; i++)
{
while (l <= r && a[q[r]] < a[i])
r--;
while (l <= r && q[l] + k <= i)
l++;
q[++r] = i;
if (i >= k)
cout << a[q[l]] << " ";
}
return 0;
}