Description
给定一个长度为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 |
你的任务是得到滑动窗口在每个位置时的最大值和最小值。
Input
输入包括两行。
第一行包括n和k,分别表示数组的长度和窗口的大小。
第二行包括n个数字。
Output
输出包括两行。
第一行包括窗口从左至右移动的每个位置的最小值。
第二行包括窗口从左至右移动的每个位置的最大值。
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
直接暴力查找.
#include <iostream>
#define SIZE 1000010
#define INF 2e+09
using namespace std;
int a[SIZE], _min[SIZE], _max[SIZE];
int main(void)
{
int n, k, i, j;
scanf("%d%d", &n, &k);
for (i = 1; i <= n; ++i)
{
scanf("%d", &a[i]);
}
for (i = k; i <= n; ++i)
{
_min[i] = INF; // 初始化最小值
_max[i] = -INF; // 初始化最大值
for (j = i - k + 1; j <= i; ++j)
{
_min[i] = min(_min[i], a[j]); // 取最小值
_max[i] = max(_max[i], a[j]); // 取最大值
}
}
for (i = k; i <= n; ++i) // 输出结果
{
printf("%d ", _min[i]);
}
printf("\n");
for (i = k; i <= n; ++i)
{
printf("%d ", _max[i]);
}
return 0;
}