优先队列(滑动窗口求最大值)

题目链接
题目中要求我们求出长度为k的滑动窗口的最大值和最小值
如果是用暴力的求法的话时间复杂度为 O ( n × k ) O(n\times k) O(n×k),这样就会导致超时。
所以我们就要优化方案,暴力的求法中的时间复杂度主要是消耗在暴力遍历寻找最大值/最小值上,因此我们要想办法优化这一段的操作。
以求最大值为例子:我们要知道一个滑动窗口中比某一个数 A A A小的并且在 A A A之前的数是对滑动窗口中的最大值是没有贡献的,因为 A A A A A A前面的这些比 A A A小的数大,还在这些数之后,就代表A会比这些数更晚出窗口,最大值永远不可能选到这些数。因此我们可以将这些数删除,也就是不考虑。所以最终滑动窗口内的所有有效数字一定是一个单调递减的数列。
假如一个滑动窗口中原本的数字分布是这样(从左往右就代表数字的先后)
在这里插入图片描述我们将无效的数字删除后就变成了这样一个单调递减的序列。
在这里插入图片描述所以我们就可以不断维护一个优先队列,每次都输出队尾元素(也就是最左边的元素)就可以。而如何维护我就在代码中进行解释。
求最小值也是一样的道理,只是反过来而已。

#include<iostream>
using namespace std;
const int N=1e6+10;

int n,k;//n数字个数,k窗口大小
int a[N],q[N];//a存原始数据,q存队列

int main(){
	cin>>n>>k;
	for(int i=0;i<n;i++) cin>>a[i];
	
	//输出最小值 
	int hh=0,tt=-1;//hh为队尾,tt为队头
	for(int i=0;i<n;i++){
		//判断队头是否已经滑出窗口
		if(hh<=tt&&i-k+1>q[hh]) hh++;//若队列内有元素,且窗口的尾端超过了队尾元素的下标就弹出队尾
		while(hh<=tt&&a[q[tt]]>=a[i]) tt--;//如果队列内有元素,且队头元素大于准备要加入的元素就将队头弹出
		q[++tt]=i;//将维护完单调性的数的下标加入队头
		if(i>=k-1) cout<<a[q[hh]]<<" ";//若窗口容量满了 
	}
	
	cout<<endl;
	
	//输出最大值 
	hh=0,tt=-1;//hh为队尾,tt为队头
	for(int i=0;i<n;i++){
		//判断队头是否已经滑出窗口
		if(hh<=tt&&i-k+1>q[hh]) hh++;//若队列内有元素,且窗口的尾端超过了队尾元素的下标就弹出队尾
		//唯一不同 <=或者>= 
		while(hh<=tt&&a[q[tt]]<=a[i]) tt--;//如果队列内有元素,且队头元素大于准备要加入的元素就将队头弹出
		q[++tt]=i;//将维护完单调性的数的下标加入队头
		if(i>=k-1) cout<<a[q[hh]]<<" ";//若窗口容量满了 
		//题目要求满了后再开始输出 
	}
	cout<<" "; 
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值