给定一个大小为 n≤106 的数组。
有一个大小为 k 的滑动窗口,它从数组的最左边移动到最右边。
你只能在窗口中看到 k 个数字。
每次滑动窗口向右移动一个位置。
以下是一个例子:
该数组为 [1 3 -1 -3 5 3 6 7],k 为 3。
窗口位置 最小值 最大值
[1 3 -1] -3 5 3 6 7 -1 3
1 [3 -1 -3] 5 3 6 7 -3 3
1 3 [-1 -3 5] 3 6 7 -3 5
1 3 -1 [-3 5 3] 6 7 -3 5
1 3 -1 -3 [5 3 6] 7 3 6
1 3 -1 -3 5 [3 6 7] 3 7
你的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。
输入格式
输入包含两行。
第一行包含两个整数 n 和 k,分别代表数组长度和滑动窗口的长度。
第二行有 n 个整数,代表数组的具体数值。
同行数据之间用空格隔开。
输出格式
输出包含两个。
第一行输出,从左至右,每个位置滑动窗口中的最小值。
第二行输出,从左至右,每个位置滑动窗口中的最大值。
输入样例:
8 3
1 3 -1 -3 5 3 6 7
输出样例:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
思路:
这道题本质是一个单调队列的问题,窗口其实就是一个不断有元素出入的队列。关于单调队列的思想与单调栈相似,下面举例说明。
对于-1来说,它在-3的“前面”,即它会先于-3出队,同时它也大于-3,因此,当当前窗口包含-1与-3时,我们并不需要考虑-1。仅针对查找最小元素,我们秉持上面的思想,尝试在窗口的移动过程中构造这么一个队列,它里面的元素一定是单调递增,队头元素最小,队尾元素最大。随着窗口的移动,一旦新进队列的元素小于队尾元素,那么我们逐个让队尾元素出队(y总原话,但这让我颇为难受,毕竟违反了传统队列的定义),相当于上面无视-1的操作,直到新元素大于当前的队尾元素,整个队列再次单调递增。而此时寻找当前窗口下的最小值
求最小值的做法明了,求最大值也清楚了。就是维护一个单调递减的队列,队头元素最大,队尾元素最小。一旦出现新元素大于队尾元素,那么逐个让队尾元素出队,直到它小于新的队尾元素。
代码
# include <iostream>
using namespace std;
const int N = 1000010;
//q为单调队列,a表示原数组,hh为队头,tt为队尾
int a[N], q[N], hh, tt = -1;
int main()
{
int n, k;
cin >> n >> k;
for (int i = 0; i < n; ++ i)
{
scanf("%d", &a[i]);
if (i - k + 1 > q[hh]) ++ hh; // 若队首出窗口,hh加1
while (hh <= tt && a[i] <= a[q[tt]]) -- tt; // 若队尾不单调,tt减1,直到新元素大于队尾元素
q[++ tt] = i; // 下标加到队尾
if (i + 1 >= k) printf("%d ", a[q[hh]]); // 输出结果
}
cout << endl;
hh = 0; tt = -1; // 重置!
for (int i = 0; i < n; ++ i)
{
if (i - k + 1 > q[hh]) ++ hh;
while (hh <= tt && a[i] >= a[q[tt]]) -- tt;
q[++ tt] = i;
if (i + 1 >= k) printf("%d ", a[q[hh]]);
}
return 0;
}
作者:Alicia编程果果
链接:https://www.acwing.com/solution/content/6564/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。