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

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

相关推荐
小白不想白a23 分钟前
每日手撕算法--哈希映射/链表存储数求和
数据结构·算法
剪一朵云爱着38 分钟前
力扣2080. 区间内查询数字的频率
算法·leetcode
落日漫游43 分钟前
数据结构笔试核心考点
java·开发语言·算法
Doro再努力44 分钟前
数据结构04:力扣顺序表3道例题解题思路与代码实现
c语言·数据结构
HY小海1 小时前
【C++】AVL树实现
开发语言·数据结构·c++
workflower1 小时前
Fundamentals of Architectural Styles and patterns
开发语言·算法·django·bug·结对编程
花月C1 小时前
高效查找数据的数据结构—MySQL 索引
数据结构·数据库·mysql
仰泳的熊猫1 小时前
LeetCode:701. 二叉搜索树中的插入操作
数据结构·c++·算法·leetcode
kali-Myon2 小时前
NewStarCTF2025-Week2-Pwn
算法·安全·gdb·pwn·ctf·栈溢出
老四啊laosi2 小时前
[双指针] 1. 力扣283.移动零
算法·leetcode·双指针·移动零