LeetCode hot100-11

java 复制代码
239. 滑动窗口最大值

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

返回 滑动窗口中的最大值 。

我的解法会超时,就比最暴力解法优化了一点点吧。能通过49/51。

思路就是左右指针中间的是窗口。记录当前窗口的最大值,窗口滑动一下,原来左指针的值出去了,新的右指针的值进来了。其实就变了两个数。就比较下新进来这个数比最大值大还是小,大的话就成为新的最大值。出去这个数比较麻烦,如果出去的是最大值的话,就要重新比较当前整个窗口找到新的最大值。就是这一步引入了第二重循环,导致会超时。

贴一下我的代码

java 复制代码
class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        if (nums.length == 0) {
            return new int[0];
        }
        int[] res = new int[nums.length - k + 1];
        int max = nums[0];
        for (int i = 0; i < k; i++) {
            if (nums[i] > max) {
                max = nums[i];
            }
        }
        res[0] = max;
        for (int i = 1; i < nums.length - k + 1; i++) {
            int j = k + i - 1;
            if (nums[j] > max) {
                max = nums[j];
            } else if (max == nums[i - 1]) {
                max = nums[i];
                for (int g = i; g <= j; g++) {
                    max = Math.max(nums[g], max);
                }
            }
            res[i] = max;

        }
        return res;

    }
}

我只看了官方解法一,困难档次的题我就不追求优化了。

解法一其实和我的想法一样,但是为了不超时,那个两重循环必须去掉。第二重循环其实就是为了找比大小。堆排序是可以降低时间复杂度的,所以就用了Java里的优先队列,用了比较器之后可以看做一个大顶堆。这个堆的每个节点是int[2]的类型,int[0]是节点的值,int[1]是节点在数组中的下标。这个下标记录下来就是为了实现如果左指针指的是滑动窗口的最大值,滑一下,就找不到最大值的这种情况。当发生这种情况时,就把堆顶移出去(也就是左指针指的那个)。移出去之后会自动构建新的大顶堆,直接就找到新的最大值了。注意这种解法堆里面的元素并不一直是k个,它里面有多少元素根本不关心,只关心当前的最大值,和最大值是否还在滑动窗口里面。所以它在移除滑动窗口之外的数据时用的while循环,因为可能第一大的节点和第二大的节点都还在堆里,但是已经在滑动窗口之外了。

大顶堆是这样定义的,Java的比较器我也不太会。记一下对于优先队列O1-O2是默认的,从小到大,O2-O1是从大到小。这里用的int[]的数据类型,用的pair2-pair1,就是从大到小。先按元素值从大到小,pair2[0] - pair1[0]。如果元素值相等,再按元素下标从大到小pair2[1] - pair1[1]

PriorityQueue<int[]> pq = new PriorityQueue<int[]>(new Comparator<int[]>() {

public int compare(int[] pair1, int[] pair2) {

return pair1[0] != pair2[0] ? pair2[0] - pair1[0] : pair2[1] - pair1[1];

}

});

再记一下PriorityQueue的几个方法,添加元素offer, 获取堆顶元素peek, 弹出堆顶元素poll.

javaclass 复制代码
    public int[] maxSlidingWindow(int[] nums, int k) {
        int n = nums.length;
        PriorityQueue<int[]> pq = new PriorityQueue<int[]>(new Comparator<int[]>() {
            public int compare(int[] pair1, int[] pair2) {
                return pair1[0] != pair2[0] ? pair2[0] - pair1[0] : pair2[1] - pair1[1];
            }
        });
        for (int i = 0; i < k; ++i) {
            pq.offer(new int[]{nums[i], i});
        }
        int[] ans = new int[n - k + 1];
        ans[0] = pq.peek()[0];
        for (int i = k; i < n; ++i) {
            pq.offer(new int[]{nums[i], i});
            while (pq.peek()[1] <= i - k) {
                pq.poll();
            }
            ans[i - k + 1] = pq.peek()[0];
        }
        return ans;
    }
}

作者:力扣官方题解
链接:https://leetcode.cn/problems/sliding-window-maximum/solutions/543426/hua-dong-chuang-kou-zui-da-zhi-by-leetco-ki6m/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
相关推荐
莫叫石榴姐24 分钟前
数据科学与SQL:组距分组分析 | 区间分布问题
大数据·人工智能·sql·深度学习·算法·机器学习·数据挖掘
茶猫_1 小时前
力扣面试题 - 25 二进制数转字符串
c语言·算法·leetcode·职场和发展
肥猪猪爸3 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
readmancynn4 小时前
二分基本实现
数据结构·算法
萝卜兽编程4 小时前
优先级队列
c++·算法
盼海4 小时前
排序算法(四)--快速排序
数据结构·算法·排序算法
一直学习永不止步4 小时前
LeetCode题练习与总结:最长回文串--409
java·数据结构·算法·leetcode·字符串·贪心·哈希表
Rstln5 小时前
【DP】个人练习-Leetcode-2019. The Score of Students Solving Math Expression
算法·leetcode·职场和发展
芜湖_5 小时前
【山大909算法题】2014-T1
算法·c·单链表
珹洺5 小时前
C语言数据结构——详细讲解 双链表
c语言·开发语言·网络·数据结构·c++·算法·leetcode