LeetCode第239题:滑动窗口最大值

LeetCode第239题

题目详述

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

返回滑动窗口中的最大值。
示例
输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]

解法一

暴力法。时间复杂度:O(Nk),空间复杂度O(N-k+1)。

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        int n = nums.length;
        if(n * k == 0) return new int[0];

        int[] output = new int[n - k + 1];
        for (int i = 0; i < n - k + 1; i++) {
            int max = Integer.MIN_VALUE;
            for (int j = i; j < i + k; j++)
                max = Math.max(max, nums[j]);
            output[i] = max;
        }
        return output;
    }
}

解法二

使用双端队列实现单调队列实现。时间复杂度:O(1),空间复杂度O(k)
时间复杂度整体上是O(1)的。

class Solution {
    class SingleQueue{
        Deque<Integer> deque = new ArrayDeque<>();
        public void push(int n){
            while (!deque.isEmpty() && deque.peekLast() < n)
                deque.pollLast();
            deque.addLast(n); 
        }
        public int max() {
            return deque.peekFirst();
        }
        public void pop(int n) {
            if (!deque.isEmpty() && deque.peekFirst() == n)
                deque.pollFirst();
        }
    }
    public int[] maxSlidingWindow(int[] nums, int k) {
        if (nums.length * k == 0) return new int[0]; 
        int[] output = new int[nums.length - k + 1];
        SingleQueue window = new SingleQueue();
        int j = 0;
        for (int i = 0; i < nums.length; i++) {
            if (i < k - 1) {
                window.push(nums[i]);//若插入的元素大于前两个元素,将新元素插入队列头
            }else {
                window.push(nums[i]);
                output[j++] = window.max();//将队列中最大值返回给结果
                window.pop(nums[i- k + 1]);//删除窗口第一个元素
            }
        }
        return output;
    }
}
发布了40 篇原创文章 · 获赞 0 · 访问量 650

猜你喜欢

转载自blog.csdn.net/weixin_42610002/article/details/104278643