算法通关村第十六关——滑动窗口与堆结合

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

复制代码
输入:nums=[1,3,-1,-3,5,3,6,7],k=3
输出:[3,3,5,5,6,7]
解释:
滑动窗口的位置                 最大值
[1 3 -1] -3 5 3 6 7            3
 1 [3 -1 -3] 5 3 6 7           3
 1 3 [-1 -3 5] 3 6 7           5
 1 3 -1 [-3 5 3] 6 7           5
 1 3 -1 -3 [5 3 6] 7           6
 1 3 -1 -3 5 [3 6 7]           7
复制代码
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;
}

优先队列中每个值存储的是一个包含元素值和对应索引的数组 [元素值, 索引]。在这段代码中,优先队列 pq 存储的是这样的数组。

pq.peek()[1] <= i - k

  1. 表示队首元素的索引值是否小于等于当前滑动窗口的起始索引。

  2. 如果队首元素的索引值小于等于当前滑动窗口的起始索引,则说明该元素已经不在当前滑动窗口的范围内,需要将其从优先队列中移除。

相关推荐
lifallen11 分钟前
Kafka 内存池MemoryPool 设计
数据结构·kafka·apache
RTC老炮29 分钟前
webrtc弱网-LossBasedBandwidthEstimation类源码分析与算法原理
网络·算法·webrtc
豆浩宇1 小时前
Conda环境隔离和PyCharm配置,完美同时运行PaddlePaddle和PyTorch
人工智能·pytorch·算法·计算机视觉·pycharm·conda·paddlepaddle
一只鱼^_1 小时前
牛客周赛 Round 108
数据结构·c++·算法·动态规划·图论·广度优先·推荐算法
小刘的AI小站2 小时前
leetcode hot100 二叉搜索树
算法·leetcode
自信的小螺丝钉2 小时前
Leetcode 876. 链表的中间结点 快慢指针
算法·leetcode·链表·指针
红豆怪怪2 小时前
[LeetCode 热题 100] 32. 最长有效括号
数据结构·python·算法·leetcode·动态规划·代理模式
愚润求学2 小时前
【贪心算法】day6
c++·算法·leetcode·贪心算法
AI 嗯啦2 小时前
计算机的排序方法
数据结构·算法·排序算法
l12345sy3 小时前
Day23_【机器学习—聚类算法—K-Means聚类 及评估指标SSE、SC、CH】
算法·机器学习·kmeans·聚类·sse·sc·ch