2022-8-23 剑指offer-优先队列（堆）-每日一题-太难不写了

剑指 Offer II 061. 和最小的 k 个数对

难度中等44收藏分享切换为英文接收动态反馈

给定两个以升序排列的整数数组 nums1 和 nums2 , 以及一个整数 k 。

定义一对值 (u,v)，其中第一个元素来自 nums1，第二个元素来自 nums2 。

请找到和最小的 k 个数对 (u1,v1),  (u2,v2)  ...  (uk,vk) 。

class Solution {
    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        PriorityQueue<int[]> q=new PriorityQueue<>((a,b)->(a[0]+a[1]-b[0]-b[1]));
        for (int i=0;i<nums1.length;i++){
            for (int j=0;j<nums2.length;j++){
                int[] temp=new int[]{nums1[i],nums2[j]};
                q.offer(temp);
            }
        }
        List<List<Integer>> ans=new ArrayList<>();
        for (int i=0;i<k;i++){
            List<Integer> list=new ArrayList<>();
            if (q.size()==0) break;
            int[] arr=q.poll();
            list.add(arr[0]);
            list.add(arr[1]);
            ans.add(list);
        }
        return ans;
    }
}

思路：优先队列。但是可以根据有序巧妙的记录所有（0，0），（1，0），。。每次取出将第二个下标+1。