题目描述
有一个长为 𝑛n 的序列 𝑎a,以及一个大小为 𝑘k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如,对于序列 [1,3,−1,−3,5,3,6,7] 以及 k=3,有如下过程:
窗口位置最小值最大值
输入格式
输入一共有两行,第一行有两个正整数 n,k。 第二行 n 个整数,表示序列 a
输出格式
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
输入输出样例
输入 #1复制
8 3 1 3 -1 -3 5 3 6 7
输出 #1复制
-1 -3 -3 -3 3 3 3 3 5 5 6 7
说明/提示
【数据范围】
对于 50% 的数据,1≤𝑛≤10 5 ,1≤n≤10 5;
对于 100% 的数据,1≤k≤n≤10 6,ai∈[−231,231)。
可以用数组或者双端队列或优先队列做
数组是最基本的
#include<iostream>
#include<algorithm>
int a[1000008],b[10000008],arr[10000008];
int l,r,n,k;
using namespace std;
void MAX()
{
l=0,r=1;
for(int i=1;i<=n;i++)
{
while(l<=r&&a[l]<=i-k)l++;
while(l<=r&&arr[i]>arr[a[r]])r--;
a[++r]=i; //求最大
if(i>=k)
cout<<arr[a[l]]<<" ";
}
}
void MIN()
{
l=0,r=1;
for(int i=1;i<=n;i++)
{
while(l<=r&&a[l]<=i-k)l++;
while(l<=r&&arr[i]<arr[a[r]])r--; //求最小
a[++r]=i;
if(i>=k)
cout<<arr[a[l]]<<" ";
}
}
int main()
{
cin>>n>>k;
for(int i=1;i<=n;i++)
{
cin>>arr[i];
}
MIN();
cout<<"\n";
MAX();
return 0;
}