取经自b站董晓算法
题目描述
有一个长为 n 的序列 a,以及一个大小为 k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如,对于序列 [1,3,−1,−3,5,3,6,7][1,3,−1,−3,5,3,6,7] 以及 k=3,有如下过程:
窗口位置最小值最大值
输入格式
输入一共有两行,第一行有两个正整数 n,k。 第二行 n 个整数,表示序列 a
输出格式
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
输入输出样例
输入
8 3 1 3 -1 -3 5 3 6 7
输出
-1 -3 -3 -3 3 3 3 3 5 5 6 7
思路:
用数组模拟队列,进行两次筛选,第一次出最小值,第二次出最大值
最小值求法:首先给元素入队,但凡出现进来的数小于队尾的数的情况,队尾出队,该数进队,永远保证队头存储最小值,当队头的下标离开滑动窗口时,队头出队,即hh++;
最大值求法同理
AC code
#include <iostream>
#include <cstring>
using namespace std;
int a[1000010];
int op[1000010];
int n, k;
int main()
{
cin >> n >> k;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
}
int tt = 0, hh = 1;
for (int i = 1; i <= n; i++)
{
while (hh <= tt && a[op[tt]] >= a[i]) tt--;
op[++tt] = i;
if (op[hh] < i - k + 1) hh++;
if (i >= k) printf("%d ", a[op[hh]]);
}
cout << endl;
memset(op, 0, sizeof op);初始化
tt = 0, hh = 1;
for (int i = 1; i <= n; i++)
{
while (hh <= tt && a[op[tt]] <= a[i]) tt--;
op[++tt] = i;
if (op[hh] < i - k + 1) hh++;
if (i >= k) printf("%d ", a[op[hh]]);
}
return 0;
}