滑动窗口

leetcode 239. 滑动窗口最大值

2019-05-27  本文已影响0人  topshi

题目描述

给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口 k 内的数字。滑动窗口每次只向右移动一位。
返回滑动窗口最大值。
相关话题: 堆、Sliding Window    难度: 困难

示例


注意:
你可以假设 k 总是有效的,1 ≤ k ≤ 输入数组的大小,且输入数组不为空。
进阶:
你能在线性时间复杂度内解决此题吗?
直观思路:

滑动窗口:
为了满足题目要求的线性时间复杂度,我们通过利用双向链表设计一种遍历规则。

例子

假设给定数组,nums = [1,3,-1,-3,-5,3,6,7], 和 k = 3


如上图,1来临,此时滑动窗口为空,直接压入;3来临,此时滑动窗口最右端为0(对应的值为1),显然3 > 1,那么将滑动窗口最右端元素弹出,将3对应的下标1压入;-3来临,-3 < -1,直接压入滑动窗口。

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if(nums.length == 0) return nums;
        LinkedList<Integer> window = new LinkedList<Integer>();
        int[] res = new int[nums.length - k + 1];
        int index = 0;
        for(int i = 0;i < nums.length;i++){
            //弹出不符合规则的元素
            while(!window.isEmpty() && 
                  nums[i] >= nums[window.peekLast()]){
                window.pollLast();
            }
            window.addLast(i);
            //清除过期元素
            if(i - window.peekFirst() == k){
                window.pollFirst();
            }
            if(i >= k - 1){
               res[index++]  = nums[window.peekFirst()];
            }
        }
        return res;
    }
}
上一篇下一篇

猜你喜欢

热点阅读