生成窗口最大值数组 + 双端队列

题目:有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置。如果数组长度为n,窗口大小为w,则一共产生n - w + 1 个窗口的最大值。

分析:

利用双端队列来实现窗口最大值的更新,从头端删除,从尾端插入和删除,属于(输入受限的双端队列),每个下标最多进队列一次,出队列一次,所以遍历过程中进出双端队列的操作的时间复杂度为O(N),整体的时间复杂度也为O(N)。首先生成双端队列qmax,队列中存放数组arr中的下标。

假设遍历到 arr[i] ,  qmax的放入规则为:

1:如果 qmax 为空, 直接放入i,放入过程结束。

2:如果qmax不为空, 取出当前队列的队尾存放的下标,假设为j,

a)如果 arr[j] > arr[i] ,直接把下标i插入队尾,放入结束。

b)如果arr[j] <= arr[i] , 把 j 从 队尾 弹出, 继续 放入规则。

加入遍历到 arr[i], qmax的弹出规则为:

如果队头的下标等于 i - w ,说明当前队头的下标已经过期,弹出队头下标即可。

代码如下:

import java.util.*;
public class Main {
		public static void main(String[]args){
			Main test = new Main() ;
			int []arr = {4 ,3 ,5 ,4 ,3 ,3 ,6 ,7} ;
			int w = 3 ;
			int []res =  test.getMaxWindow(arr, w) ;
			System.out.println(Arrays.toString(res));
		
  }
		
		public int[] getMaxWindow(int [] arr , int w){
			if(arr == null || w < 1 || arr.length < w){
				return null ;
			}
			LinkedList<Integer> qmax = new LinkedList<Integer>() ;
			int [] res = new int[arr.length - w + 1] ;
			int index = 0 ;
			for(int i = 0 ; i<arr.length ; i++){
				while(!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[i]){
					qmax.pollLast() ; //队尾弹出较小值的下标
				}
				qmax.addLast(i); //队尾放入下标
				if(qmax.peekFirst() == i - w){  // 队头弹出过期的下标
					qmax.pollFirst();
				}
				if(i >= w - 1){ //从 w -1 位开始保存窗口最大值
					res[index++] = arr[qmax.peekFirst()] ;
				}
			}
			return res ;
		}		
}


 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值