总时间限制: 12000ms
内存限制: 65536kB
描述
给定一个长度为n(n<=10^6)的数组。有一个大小为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个数字。
扫描二维码关注公众号,回复:
12274752 查看本文章
输出
输出包括两行。
第一行包括窗口从左至右移动的每个位置的最小值。
第二行包括窗口从左至右移动的每个位置的最大值。
样例输入
8 3 1 3 -1 -3 5 3 6 7
样例输出
-1 -3 -3 -3 3 3 3 3 5 5 6 7
#include <iostream>
#include <cstring>
#include <algorithm>
#include <climits>
#include <deque>
using namespace std;
const int MAXN = 1e6;
int n, k;
int a[MAXN+5];
int main()
{
scanf("%d%d", &n, &k);
for (int i = 0; i < n; ++i)
scanf("%d", &a[i]);
/* min */
deque<int> q;
for (int i = 0; i < k-1; ++i) {
while (!q.empty() && a[q.front()] > a[i])
q.pop_front();
q.push_front(i);
}
for (int i = k-1; i < n; ++i) {
if (!q.empty() && i - q.back() >= k)
q.pop_back();
while (!q.empty() && a[q.front()] > a[i])
q.pop_front();
q.push_front(i);
printf("%d ", a[q.back()]);
}
printf("\n");
/* max */
q.clear();
for (int i = 0; i < k-1; ++i) {
while (!q.empty() && a[q.front()] < a[i])
q.pop_front();
q.push_front(i);
}
for (int i = k-1; i < n; ++i) {
if (!q.empty() && i - q.back() >= k)
q.pop_back();
while (!q.empty() && a[q.front()] < a[i])
q.pop_front();
q.push_front(i);
printf("%d ", a[q.back()]);
}
printf("\n");
system("pause");
return 0;
}
【分析】
单调队列问题,十分巧妙
可以看看这篇题解