题目描述:
有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置,求每一种窗口状态下的最大值。(如果数组长度为n,窗口大小为w,则一共产生n-w+1个窗口的最大值)
输入描述:
第一行输入 n 和 w ,分别代表数组长度和窗口大小
第二行输入 n 个整数 X i X_i Xi ,表示数组中的各个元素
输出描述:
输出一个长度为 n-w+1 的数组 res,res[i] 表示每一种窗口状态下的最大值
示例1
输入
8 3
4 3 5 4 3 3 6 7
输出
5 5 5 4 6 7
说明
例如,数组为[4,3,5,4,3,3,6,7],窗口大小为3时:
[4 3 5] 4 3 3 6 7 窗口中最大值为5
4 [3 5 4] 3 3 6 7 窗口中最大值为5
4 3 [5 4 3] 3 6 7 窗口中最大值为5
4 3 5 [4 3 3] 6 7 窗口中最大值为4
4 3 5 4 [3 3 6] 7 窗口中最大值为6
4 3 5 4 3 [3 6 7] 窗口中最大值为7
输出的结果为{5 5 5 4 6 7}
备注:
1 ≤ w ≤ n ≤ 1 0 6 1 \leq w \leq n \leq 10^6 1≤w≤n≤106
− 1 0 6 ≤ X i ≤ 1 0 6 -10^6 \leq X_i \leq 10^6 −106≤Xi≤106
题解:
用一个单调递减双端队列维护大小为 w 的窗口最大值,设当前元素为 cur ,若 cur 小于队尾元素,则将 cur 插入队尾;否则的话,从队尾中弹出小于等于 cur 的元素,维护队列的单调性。
代码:
#include <cstdio>
using namespace std;
const int N = 1000010;
int a[N];
int dq[N];
int main(void) {
int n, w, x;
scanf("%d%d", &n, &w);
int h = 0, t = 0;
for (int i = 0; i < n; ++i) {
scanf("%d", a + i);
if (t < h && i - dq[t] == w) ++t;
while (t < h && a[dq[h - 1]] <= a[i]) --h;
dq[h++] = i;
if (i >= w - 1) printf("%d%c", a[dq[t]], " \n"[i == n - 1]);
}
return 0;
}