题意:
有一个长度为 n 的数列和一个大小为 k 的窗口,窗口可以在数列上来回移动。求:窗口在从左往右滑的时候,每次窗口内数的最大值和最小值分别是多少。
例如下表中,数列是 [1 3 -1 -3 5 3 6 7],其中 k 等于 3。
Window position | Minimum value | Maximum value |
---|---|---|
[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 | 6 |
输入要求:
输入两行,第一行两个整数 n 和 k 分别表示数列长度和滑动窗口的大小,1<=k<=n<=1000000。第二行输入 n 个整数表示数列。
输出要求:
输出两行,第一行输出滑动窗口在从左到右的每个位置时,滑动窗口中的最小值,第二行输出最大值。
Sample Input:
8 3
1 3 -1 -3 5 3 6 7
Sample Output:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
思路:
- 首先想到暴力解法,枚举所有窗口的情况,显然,复杂度过高,接近 O( n 2 n^2 n2),所以另寻他法。
- 这里使用的单调队列,通过单调递增队列和单调递减队列,分别求出每个窗口中的最小值和最大值。
- 从数组第一位开始,先将 k 位数插入单调队列中,并将队首的值记录,然后从第 k+1 位开始,每次从队尾插入一个元素,再从队首删除一个元素,记录队首,直到整个数列全部插入。
- 队列用数组 a 和队首 front 和队尾 back 进行记录。单调队列则体现在向队列中插入数据时,先比较队尾元素和插入元素大小再进行删除或插入操作。
代码:
#include <iostream>
using namespace std;
pair<int, int> a[1000005];
pair<int, int> windows[1000005];
int maxAns[1000005];
int minAns[1000005];
int back = -1, front = 0;
void push_back_up(pair<int, int> p) {//单调递增队列
if (back < front) {
back++;
windows[back] = p;
return;
}
while (1) {
if (back < front) {//若删空了队列
back++;
windows[back] = p;
return;
}
if (p.second > windows[back].second) {
back++;
windows[back] = p;
break;
}
else {
back--;
}
}
}
void push_back_down(pair<int, int> p) {//单调递减队列
if (back < front) {
back++;
windows[back] = p;
return;
}
while (1) {
if (back < front) {//若删空了队列
back++;
windows[back] = p;
return;
}
if (p.second < windows[back].second) {
back++;
windows[back] = p;
break;
}
else {
back--;
}
}
}
int main()
{
int n, k;
scanf_s("%d%d", &n, &k);
for (int i = 1; i <= n; i++) {
a[i].first = i;
scanf_s("%d", &a[i].second);
}
//求各个区间最小值
int i = 1;
for (i; i <= k - 1; i++) {
push_back_up(a[i]);
}
while (i <= n) {
push_back_up(a[i]);
if (windows[front].first < i - k + 1) {
front++;
}
minAns[i - k] = windows[front].second;
i++;
}
for (int i = 0; i < n - k + 1; i++) {
printf("%d ", minAns[i]);
}
printf("\n");
//求各个区间最大值
back = -1;
front = 0;
i = 1;
for (i; i <= k - 1; i++) {
push_back_down(a[i]);
}
while (i <= n) {
push_back_down(a[i]);
if (windows[front].first < i - k + 1) {
front++;
}
maxAns[i - k] = windows[front].second;
i++;
}
for (int i = 0; i < n - k + 1; i++) {
printf("%d ", maxAns[i]);
}
}