0
点赞
收藏
分享

微信扫一扫

I.滑动窗口

殇感故事 2022-02-19 阅读 59
c++

题意

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

相关推荐

0 条评论