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

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

相关推荐
学不动CV了几秒前
ARM单片机OTA解析(二)
arm开发·数据结构·stm32·单片机·嵌入式硬件
kk在加油16 分钟前
Redis基础数据结构
数据结构·数据库·redis
平和男人杨争争17 分钟前
机器学习12——支持向量机中
算法·机器学习·支持向量机
10岁的博客34 分钟前
代码编程:一场思维与创造力的革命
开发语言·算法
IT永勇1 小时前
数据结构-栈
c语言·数据结构·嵌入式开发
Aczone281 小时前
嵌入式 数据结构学习 (六) 树、哈希表与内核链表
数据结构·学习·算法
定偶1 小时前
进制转换小题
c语言·开发语言·数据结构·算法
体系结构论文研讨会2 小时前
多项式环及Rq的含义
算法
智驱力人工智能2 小时前
极端高温下的智慧出行:危险检测与救援
人工智能·算法·安全·行为识别·智能巡航·高温预警·高温监测
森焱森2 小时前
60 美元玩转 Li-Fi —— 开源 OpenVLC 平台入门(附 BeagleBone Black 驱动简单解析)
c语言·单片机·算法·架构·开源