题目描述 Description
给你一个长度为N的数组,一个长为K的滑动的窗体从最左移至最右端,你只能见到窗口的K个数,每次窗体向右移动一位,如下表:
Window position | Min value | Max 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 | 7 |
你的任务是找出窗口在各位置时的max value, min value.
输入描述 Input Description
第1行n,k,第2行为长度为n的数组
输出描述 Output Description**
2行
第1行每个位置的min value
第2行每个位置的max value
样例输入 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
数据范围及提示 Data Size & Hint
数据范围:20%: n<=500; 50%: n<=100000;100%: n<=1000000;
题解 Solution
对于此题,我们可以维护两个个单调队列(一个单调递增,一个单调递减):
当窗口滑动时,首先要删除队列的最左边元素(如果是有用元素),然后把新元素加入队列。注意,当新元素加入队列时,比新元素大(小)的都变得无用了,应当从右边往左边删除,直到下一个元素比新元素小(大)为止。
因此,我们可以用到STL中的deque来实现。
- 代码
#include <cstdio>
#include <cstdlib>
#include <deque>
using namespace std;
const int maxn = (int)1e6 + 5;
struct point {
int a, t;
};
deque<point> dmin, dmax;
int a[maxn], Min[maxn], Max[maxn];
int n, k;
int main()
{
freopen("input.in", "r", stdin);
freopen("output.out", "w", stdout);
int i;
scanf("%d%d", &n, &k);
for(i = 1; i <= n; i++)
scanf("%d", &a[i]);
for(i = 1; i <= n; i++) {
while(!dmax.empty() && dmax.back().a <= a[i])
dmax.pop_back();
dmax.push_back((point){a[i], i});
while(!dmin.empty() && dmin.back().a >= a[i])
dmin.pop_back();
dmin.push_back((point){a[i], i});
if(i >= k) {
Min[i - k + 1] = dmin.front().a; Max[i - k + 1] = dmax.front().a;
if(dmin.front().t <= i - k + 1) dmin.pop_front();
if(dmax.front().t <= i - k + 1) dmax.pop_front();
}
}
for(i = 1; i <= n - k + 1; i++)
printf("%d ", Min[i]);
printf("\n");
for(i = 1; i <= n - k + 1; i++)
printf("%d ", Max[i]);
return 0;
}
注:虽然不知道为什么,此程序交到poj上会超时