Java实现滑动窗口

滑动窗口

问题

LeetCode:239. 滑动窗口最大值
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回滑动窗口中的最大值

窗口

这里窗口采用双端队列实现(即双向链表),并且窗口必须满足从大到小排列,如下所示
向窗口添加数据时:

假如窗口大小为8,遍历数组,并依次放入队列,存在三种可能:

  1. 当当前数小于队列末尾数时,满足 (大->小) ,则添加到队尾,如:队列状态2;
  2. 当当前数大于队列末尾数时,不满足 (大->小) ,则弹出队尾数,并继续向前比较直到存在比当前数大或者队列为空时,再将当前数添加至队尾,如:队列状态5,7大于之前所有的数;
  3. 当当前数等于队列末尾数时,弹出队尾数,将当前数添加进队列
原数组54127596
窗口状态15
窗口状态254
窗口状态3541
窗口状态4542
窗口状态57
窗口状态675
窗口状态79
窗口状态896

窗口移动时:

记录窗口左边界,从窗口左边弹出下标不在边界内的数据,并同时向窗口右边添加数据,直到窗口越界
假如窗口大小为3

原数组54127596
窗口框住的数据541
窗口状态1541
窗口框住的数据412
窗口状态242
窗口框住的数据127
窗口状态37
窗口框住的数据275
窗口状态475
窗口框住的数据759
窗口状态59
窗口框住的数据596
窗口状态696

Java代码

注意:代码实现中,双端队列中存的是数组值的下标,并不是值本身

	public static int[] maxSlidingWindow(int[] arr,int k) {
		if(arr.length == 0 || arr.length == 1)return arr;
		int[] res = new int [arr.length - k + 1];
		LinkedList<Integer> list = new LinkedList<>();
		for (int i = 0; i < arr.length; i++) {	
			//添加数据时,满足2、3状态时,弹出队尾
			while(!list.isEmpty() && arr[list.peekLast()] <= arr[i]) {
				list.pollLast();
			}
			list.addLast(i);
			//队头下标越界弹出队头
			if(i - k >= list.peekFirst()) {
				list.pollFirst();
			}
			if(i + 1 >= k) {	
				res[i - k + 1] = arr[list.peekFirst()];
			}
		}
		return res;
	}

进阶版题目

最大值减去最小值小于或等于num的子数组数量
题目:
给定数组arr和整数num,共返回有多少个子数组满足如下情况:
max(arr【i…j】)-min(ar[i…j]) <= num
要求:
如果数组长度为N,情实现时间复杂度为O(N)的解法

public static int count(int[] arr, int num) {
		int L = 0;
		int R = 0;
		LinkedList<Integer> max = new LinkedList<>();
		LinkedList<Integer> min = new LinkedList<>();
		int res = 0;
		while(L < arr.length) {
			//窗口右边界向右扩,直到不满足最大值最小值差大于num
			while(R < arr.length) {
				//最大值链表由大到小存好
				while(!max.isEmpty() && arr[max.peekLast()] <= arr[R]) {
					max.pollLast();
				}
				max.addLast(R);
				//最小值链表由小到大存好
				while(!min.isEmpty() && arr[min.peekLast()] >= arr[R]) {
					min.pollLast();
				}
				min.addLast(R);
				//不满足条件则break
				if(arr[max.peekFirst()] - arr[min.peekFirst()] > num) {
					break;
				}
				R++;
			}
			//添加当前窗口内存在以左边界为起始的序列数共有R-L个
			res += R - L;
			//弹出最大值或最小值为左边界的数,因为左边界要自增一
			if(max.peekFirst() == L) {
				max.pollFirst();
			}
			if(min.peekFirst() == L){
				min.pollFirst();
			}
			L++;
		}
		return res;
	}

总结

一般求连续区间的问题时,可以用滑动窗口

  • 5
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值