问题描述#
给定一个大小为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 -1 -3 5 3 6 7
第一步 1 先入队了
第二步 1 小于3 没问题 所以3接着入队了
第三步 1和3 都大于-1 所以1和3都弹出 -1入队 此时刚好满足长度为k(k=3)的滑动窗口,输出队首 -1
第四步 -1 大于 -3 所以-1弹出 -3 入队 输出队首-3
第五步 -3 小于5 没问题 5入队 输出队首-3
第六步 5大于3 那5不论怎么样都不可能是最小值了,所以弹出5 ,3入队,输出队首-3
第六步 3小于6没问题 6入队,此时滑动窗口覆盖范围已经走到了(5,3,6) 所以要将-3 出队,输出队首3
第七步 6小于7没问题 ,7入队,输出队首3
求最大值的话同理
代码#
#include<bits/stdc++.h>
#include<stdio.h>
using namespace std;
typedef long long ll;
#define readc(x) scanf("%c",&x)
#define read(x) scanf("%d",&x)
#define read2(x,y) scanf("%d%d",&x,&y)
#define read3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define print(x) printf("%d\n",x)
#define mst(a,b) memset(a,b,sizeof(a))
#define pb push_back
#define mp make_pair
const int maxn = 1e6 + 5;
int a[maxn];
struct node {
int v; //值
int index ; //下标
}que[maxn];
int main(){
int n,k;
read2(n,k);
for(int i = 1 ; i <= n; i++){
read(a[i]);
}
int head = 1;
int tail = 0;
//求队列中窗口最小值
for(int i = 1 ; i <= n; i++){
while(head <= tail && que[tail].v >= a[i])
tail--; //如果队列中的数大于要入队的数,就全补弹出
que[++tail].v = a[i]; que[tail].index = i; //入队
if(i - k >= que[head].index) head++; //超出窗口的长度,弹出队首
if(i >= k) { // 前k-1个无法形成滑动窗口
printf("%d",que[head].v);
if(i != n) printf(" ");
}
}
printf("\n");
head = 1,tail = 0;
//求队列中窗口最大值
for(int i = 1 ; i <= n; i++){
while(head <= tail && que[tail].v <= a[i]) tail--;
que[++tail].v = a[i]; que[tail].index = i;
if(i - k >= que[head].index) head++;
if(i >= k) {
printf("%d",que[head].v);
if(i != n) printf(" ");
}
}
printf("\n");
}