题意
给一个长度为 NN 的数组,一个长为 KK 的滑动窗体从最左端移至最右端,你只能看到窗口中的 KK 个数,每次窗体向右移动一位,如下图:
|窗口位置|最小值|最大值|
|:-:|:-:|:-:|
|\texttt{[1 3 -1] -3 5 3 6 7}[1 3 -1] -3 5 3 6 7|-1−1|33|
|\ \texttt{ 1 [3 -1 -3] 5 3 6 7} 1 [3 -1 -3] 5 3 6 7|-3−3|33|
|\ \texttt{ 1 3 [-1 -3 5] 3 6 7} 1 3 [-1 -3 5] 3 6 7|-3−3|55|
|\ \texttt{ 1 3 -1 [-3 5 3] 6 7} 1 3 -1 [-3 5 3] 6 7|-3−3|55|
|\ \texttt{ 1 3 -1 -3 [5 3 6] 7} 1 3 -1 -3 [5 3 6] 7|33|66|
|\ \texttt{ 1 3 -1 -3 5 [3 6 7]} 1 3 -1 -3 5 [3 6 7]|33|77|
你的任务是找出窗体在各个位置时的最大值和最小值。
输入
第 1 行:两个整数 NN 和 KK;
第 2 行:NN 个整数,表示数组的 NN 个元素(≤2\times 10^9≤2×109);
输出
第一行为滑动窗口从左向右移动到每个位置时的最小值,每个数之间用一个空格分开;
第二行为滑动窗口从左向右移动到每个位置时的最大值,每个数之间用一个空格分开。
样例
8 3 -1 -3 -3 -3 3 3 1 3 -1 -3 5 3 6 7 3 3 5 5 6 7
思路
对于每次的一次窗口,如果每滑动一次,就要从头到尾计算窗口的最大值或者最小值,那么时间的消耗过大。可以发现,每次对于一个窗口,都有一个确定的最大值或者最小值,然后也可以得知第二个大和第二小的值,当这个最大值没有离开滑动窗口,并且滑动窗口新加入的值没有这个最大值大,那么这时候的滑动窗口的最大值都是确定的,当最大值滑出窗口或者新加入了比他还大的值,那么这时候窗口的值就改变了,对于前者就是第二大的值变成最大值,对于后者就是新加入的值变成最大值,原先的最大值变成第二大值,那么对于这个过程我们只需要维护一个单调队列即可,对于最小值也是同理,所以就是维护一个单调队列。
代码
#include<stdio.h>
#define ll long long
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
struct stu
{
ll ans;
ll NO;
};
ll que[10000008];
ll MAX[10000008];
ll MIN[10000008];
struct stu a[10000008];
struct stu b[10000008];
int main()
{
ll N,K;
scanf("%lld %lld",&N,&K);
for(ll i=1;i<=N;i++)
scanf("%lld",&que[i]);
ll head=1,tail=0;
for(ll i=1;i<K;i++)
{
while( que[i] <=a[tail].ans && head<=tail )
tail--;
a[++tail].ans=que[i];
a[tail].NO=i;
}
for(ll i=K;i<=N;i++)
{
while( que[i] <=a[tail].ans && head<=tail )
tail--;
a[++tail].ans=que[i];
a[tail].NO=i;
while(a[head].NO <i-K+1)
head++;
MIN[i-K+1]=a[head].ans;
}
head=1,tail=0;
for(ll i=1;i<K;i++)
{
while( que[i] >=b[tail].ans && head<=tail )
tail--;
b[++tail].ans=que[i];
b[tail].NO=i;
}
for(ll i=K;i<=N;i++)
{
while( que[i] >=b[tail].ans && head<=tail )
tail--;
b[++tail].ans=que[i];
b[tail].NO=i;
while(b[head].NO <i-K+1)
head++;
MAX[i-K+1]=b[head].ans;
}
for(ll i=1;i<=N-K+1;i++)
{
printf("%lld ",MIN[i]);
}
printf("\n");
for(ll i=1;i<=N-K+1;i++)
{
printf("%lld ",MAX[i]);
}
return 0;
}