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

void1024

于 2024-07-17 00:58:23 发布

阅读量307

点赞数 3

分类专栏：面试经典 150 题文章标签：算法 leetcode 数据结构 java

本文链接：https://blog.csdn.net/diOSyu/article/details/140480810

版权

面试经典 150 题专栏收录该内容

132 篇文章 1 订阅

订阅专栏

class Solution {
    private int[] arr1, arr2;

    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        arr1 = nums1;
        arr2 = nums2;
        List<List<Integer>> ans = new ArrayList<>(k);
        PriorityQueue<int[]> Q = new PriorityQueue<>(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return (arr1[o1[0]] + arr2[o1[1]]) - (arr1[o2[0]] + arr2[o2[1]]);
            }
        });
        for (int i = 0; i < arr1.length; i++) {
            Q.add(new int[]{i, 0});
        }
        for (int i = 0; i < k; i++) {
            int[] poll = Q.poll();
            ans.add(Arrays.asList(arr1[poll[0]], arr2[poll[1]]));
            if (poll[1] + 1 < arr2.length) {
                Q.add(new int[]{poll[0], poll[1] + 1});
            }
        }
        return ans;
    }
}