力扣面试150 查找和最小的 K 对数字 最小堆 去重

Problem: 373. 查找和最小的 K 对数字

👨‍🏫 参考题解

Java 复制代码
class Solution {
    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        // 创建一个大小为 k 的结果列表,用于存储和最小的 k 个数对
        List<List<Integer>> ans = new ArrayList<>(k); // 预分配空间
        
        // 创建一个优先队列(小根堆),存储三元组 [nums1[i] + nums2[j], i, j]
        // 按照和 (nums1[i] + nums2[j]) 的大小升序排列
        PriorityQueue<int[]> pq = new PriorityQueue<>((a, b) -> a[0] - b[0]);
        
        // 将 nums1 中前 k 个元素与 nums2 中第一个元素的和及其索引 i, j 加入到优先队列中
        for (int i = 0; i < Math.min(nums1.length, k); i++) { // 至多 k 个
            pq.add(new int[]{nums1[i] + nums2[0], i, 0});
        }
        
        // 循环直到找到 k 个数对或者优先队列为空
        while (ans.size() < k && !pq.isEmpty()) {
            // 取出堆顶元素,也就是当前和最小的数对
            int[] p = pq.poll();
            int i = p[1]; // 取出 nums1 的索引
            int j = p[2]; // 取出 nums2 的索引
            
            // 将当前和最小的数对加入结果列表
            ans.add(List.of(nums1[i], nums2[j]));
            
            // 如果 nums2 中还有剩余元素,将新的数对 [nums1[i], nums2[j + 1]] 放入优先队列
            if (j + 1 < nums2.length) {
                pq.add(new int[]{nums1[i] + nums2[j + 1], i, j + 1});
            }
        }
        
        // 返回结果列表
        return ans;
    }
}
相关推荐
Swift社区11 分钟前
LeetCode 399 除法求值
算法·leetcode·职场和发展
仰泳的熊猫19 分钟前
LeetCode:98. 验证二叉搜索树
数据结构·c++·算法·leetcode
JAVA学习通32 分钟前
零基础OSS组件(Java)
java·linux·leetcode
Python智慧行囊32 分钟前
图像处理(三)--开运算与闭运算,梯度运算,礼帽与黑帽
人工智能·算法·计算机视觉
前端小L34 分钟前
动态规划的“细节魔鬼”:子序列 vs 子数组 —— 最长重复子数组
算法·动态规划
草莓熊Lotso36 分钟前
《算法闯关指南:优选算法--二分查找》--19.x的平方根,20.搜索插入位置
java·开发语言·c++·算法
sali-tec42 分钟前
C# 基于halcon的视觉工作流-章46-不匀面划痕
人工智能·算法·计算机视觉·c#
绝无仅有1 小时前
面试真实经历某商银行大厂Java问题和答案总结(四)
后端·面试·github
绝无仅有1 小时前
面试真实经历某商银行大厂Java问题和答案总结(六)
后端·面试·github
yuniko-n2 小时前
【力扣 SQL 50】连接
数据库·后端·sql·算法·leetcode