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;

        
    }
}
相关推荐
麟城Lincoln34 分钟前
【RHCSA-Linux考试题目笔记(自用)】servera的题目
linux·笔记·考试·rhcsa
寻月隐君1 小时前
保姆级教程:Zsh + Oh My Zsh 终极配置,让你的 Ubuntu 终端效率倍增
linux·后端·命令行
XM-54581 小时前
2025微信小程序wxapkg解包全攻略
linux·运维·小程序
三维重建-光栅投影1 小时前
VS中将cuda项目编译为DLL并调用
算法
朗晴2 小时前
文本编辑器VIM的使用方法!
linux·运维·服务器
课堂剪切板4 小时前
ch03 部分题目思路
算法
山登绝顶我为峰 3(^v^)35 小时前
如何录制带备注的演示文稿(LaTex Beamer + Pympress)
c++·线性代数·算法·计算机·密码学·音视频·latex
Two_brushes.6 小时前
【算法】宽度优先遍历BFS
算法·leetcode·哈希算法·宽度优先
森焱森8 小时前
水下航行器外形分类详解
c语言·单片机·算法·架构·无人机
2401_826097629 小时前
JavaEE-Linux环境部署
java·linux·java-ee