#47. 滑行的窗口
统计
描述
提交
自定义测试
【题目描述】:
给定一个长度为n的数列a,再给定一个长度为k的滑动窗口,从第一个数字开始依次框定k个数字,求每次框定的数字中的最大值和最小值,依次输出所有的这些值。下面有一个例子数组是 [1 3 1 3 5 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
【时间限制、数据范围及描述】:
时间:1s 空间:64M
30%:n<=100 k<=20
60%:n<=5000 k<=20
100%:n<=10^6,每个元素不操过int类型
需要读入输出优化
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <stack>
#include <cmath>
using namespace std;
const int MAXN=1000005;
struct cyq{
int num,ss;
};
int n,k,tt,h1=1,h2=1,t1=1,t2=1,a[MAXN];
cyq q1[MAXN],q2[MAXN];
inline int get()
{
int data=0,w=1; char ch=0;
while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
if(ch=='-') w=-1,ch=getchar();
while(ch>='0' && ch<='9') data=data*10+ch-'0',ch=getchar();
return data*w;
}
void work(){
tt=get();a[1]=tt;
q1[1].num=1;q1[1].ss=tt;q2[1].num=1;q2[1].ss=tt;
if(k==1) printf("%d ",tt);
for(int i=2;i<=n;i++){
tt=get();a[i]=tt;
while(i-q1[h1].num+1>k) h1++;
while(q1[t1].ss>tt && t1>=h1) t1--;
q1[++t1].num=i;q1[t1].ss=tt;
if(i>=k) printf("%d ",q1[h1].ss);
}printf("\n");
if(k==1) printf("%d ",a[1]);
for(int i=2;i<=n;i++){
tt=a[i];
while(i-q2[h2].num+1>k) h2++;
while(q2[t2].ss<tt && t2>=h2) t2--;
q2[++t2].num=i;q2[t2].ss=tt;
if(i>=k) printf("%d ",q2[h2].ss);
}printf("\n");
}
int main(){
n=get();k=get();
work();
return 0;
}