算法通关村——解析堆的应用

在数组中找第K大的元素

LeetCode21 Medium

我们要找第 K 大的元素,如果我们找使用大堆的话那么就会造成这个堆到底需要多大的,而且哪一个是第 K 的的元素我们不知道是哪一个索引,我们更想要的结果就是根节点就是我们要找的值,所以我们可以使用 小堆,使用小堆的好处就是,我们可以用到小堆的性质:根节点最小。使用这个我们在结合 if 判断一下,就可以实现这个效果了!

java 复制代码
import java.util.PriorityQueue;
public class Solution {
    public int findKthLargest(int[] nums, int k) {
        if(k>nums.length){
            return -1;
        }
        int len = nums.length;
        // 使用一个含有 k 个元素的最小堆
        PriorityQueue<Integer> minHeap = new PriorityQueue<>(k, (a, b) -> a - b);
        for (int i = 0; i < k; i++) {
            minHeap.add(nums[i]);
        }
        for (int i = k; i < len; i++) {
            // 看一眼,不拿出,因为有可能没有必要替换
            Integer topEle = minHeap.peek();
            // 只要当前遍历的元素比堆顶元素大,堆顶弹出,遍历的元素进去
            if (nums[i] > topEle) {
                minHeap.poll();
                minHeap.offer(nums[i]);
            }
        }
        return minHeap.peek();
    }
}

小结一下:

  1. K多大就建立多大固定大小的堆
  2. 找最大用小堆,
  3. 只有比根元素大的才让进入堆。

合并K个排序链表

合并K个排序链表 Hard

priorityQueue.offer(tail.next) 这个操作保证了合并后的链表也是有序的

java 复制代码
Class solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) {
            return null;
        }
        // 创建一个最小堆
        PriorityQueue<ListNode> priorityQueue = new PriorityQueue<>(Comparator.comparing(node -> node.val));
        for (ListNode list : lists) {
            if (list != null) {
                priorityQueue.add(list);
            }
        }
        // 记录头节点
        ListNode dummy = new ListNode(0);
        ListNode tail = dummy;
        // 进行排序
        while (!priorityQueue.isEmpty()) {
            tail.next = priorityQueue.poll();
            tail = tail.next;
            if (tail.next != null) {
                priorityQueue.offer(tail.next);
            }
        }
        return dummy.next;
    }
}
相关推荐
✿ ༺ ོIT技术༻12 分钟前
剑指offer第2版:链表系列
数据结构·算法·链表
yiridancan36 分钟前
终极剖析HashMap:数据结构、哈希冲突与解决方案全解
java·数据结构·算法·哈希算法
满分观察网友z38 分钟前
性能优化大作战:从 O(N*M) 到 O(N),我的哈希表奇遇记(1865. 找出和为指定值的下标对)
算法
点云SLAM3 小时前
二叉树算法详解和C++代码示例
数据结构·c++·算法·红黑树·二叉树算法
今天背单词了吗98010 小时前
算法学习笔记:19.牛顿迭代法——从原理到实战,涵盖 LeetCode 与考研 408 例题
笔记·学习·算法·牛顿迭代法
jdlxx_dongfangxing10 小时前
进制转换算法详解及应用
算法
why技术11 小时前
也是出息了,业务代码里面也用上算法了。
java·后端·算法
2501_9228955812 小时前
字符函数和字符串函数(下)- 暴力匹配算法
算法
IT信息技术学习圈13 小时前
算法核心知识复习:排序算法对比 + 递归与递推深度解析(根据GESP四级题目总结)
算法·排序算法
愚润求学13 小时前
【动态规划】01背包问题
c++·算法·leetcode·动态规划