Leetcode 查找和最小的 K 对数字

java 实现

java 复制代码
class Solution {
    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        //首先创建一个存储结果的二维数组
        List<List<Integer>> result = new ArrayList<>();
        //特殊情况处理
        if(nums1 == null || nums2 == null || nums1.length == 0 || nums2.length == 0) {
            return result;
        }

        //然后初始化优先队列, 
        //其中a[0] b[0]是数对中第一个数字在数组 nums1 中的索引, a[1] b[1]是数对中第二个数字在数组 nums2 中的索引
        PriorityQueue<int[]> minHeap = new PriorityQueue<>((a, b) -> (nums1[a[0]] + nums2[a[1]]) - (nums1[b[0]] + nums2[b[1]]));
        //初始化堆,先让 nums1 的前 k 个元素和 nums2[0] 配对加入堆
        for(int i = 0; i < Math.min(nums1.length, k); i++) {
            minHeap.offer(new int[]{i, 0});
        }

        //然后取出前 k 小的数对
        while(k-- > 0 && !minHeap.isEmpty()) {
            //堆顶元素出队
            int[] pair = minHeap.poll();
            int i = pair[0], j = pair[1];

            List<Integer> currentPair = new ArrayList<>();
            currentPair.add(nums1[i]);
            currentPair.add(nums2[j]);
            result.add(currentPair);

            //如果 nums2 中还有下一个元素, 加入堆
            //这一部分代码片段在往堆中添加元素时,会根据最小堆设定的规则动态调整堆的堆顶元素位置
            //所以上面初始化堆时,先让 nums1 的前 k 个元素和 nums2[0] 配对加入堆也会得到正确的结果。
            if(j + 1 < nums2.length) {
                minHeap.offer(new int[]{i, j + 1});
            }
        }

        return result;

        
    }
}
相关推荐
小小小小王王王7 分钟前
求猪肉价格最大值
数据结构·c++·算法
岁忧31 分钟前
(LeetCode 面试经典 150 题 ) 58. 最后一个单词的长度 (字符串)
java·c++·算法·leetcode·面试·go
BIYing_Aurora38 分钟前
【IPMV】图像处理与机器视觉:Lec13 Robust Estimation with RANSAC
图像处理·人工智能·算法·计算机视觉
即将头秃的程序媛2 小时前
centos 7.9安装tomcat,并实现开机自启
linux·运维·centos
fangeqin2 小时前
ubuntu源码安装python3.13遇到Could not build the ssl module!解决方法
linux·python·ubuntu·openssl
martian6652 小时前
支持向量机(SVM)深度解析:从数学根基到工程实践
算法·机器学习·支持向量机
孟大本事要学习2 小时前
算法19天|回溯算法:理论基础、组合、组合总和Ⅲ、电话号码的字母组合
算法
??tobenewyorker3 小时前
力扣打卡第二十一天 中后遍历+中前遍历 构造二叉树
数据结构·c++·算法·leetcode
让我们一起加油好吗3 小时前
【基础算法】贪心 (二) :推公式
数据结构·数学·算法·贪心算法·洛谷
爱奥尼欧3 小时前
【Linux 系统】基础IO——Linux中对文件的理解
linux·服务器·microsoft