题意
给一个长度为 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;
}