给定两个以升序排列的整形数组 nums1 和 nums2, 以及一个整数 k。
定义一对值 (u,v)，其中第一个元素来自 nums1，第二个元素来自 nums2。
找到和最小的 k 对数字 (u1,v1), (u2,v2) ... (uk,vk)。
示例 1:
输入: nums1 = [1,7,11], nums2 = [2,4,6], k = 3
输出: [1,2],[1,4],[1,6]
解释: 返回序列中的前 3 对数：
[1,2],[1,4],[1,6],[7,2],[7,4],[11,2],[7,6],[11,4],[11,6]
示例 2:
输入: nums1 = [1,1,2], nums2 = [1,2,3], k = 2
输出: [1,1],[1,1]
解释: 返回序列中的前 2 对数：
[1,1],[1,1],[1,2],[2,1],[1,2],[2,2],[1,3],[1,3],[2,3]
示例 3:
输入: nums1 = [1,2], nums2 = [3], k = 3
输出: [1,3],[2,3]
解释: 也可能序列中所有的数对都被返回:[1,3],[2,3]
答案：
1public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
2 PriorityQueue<int[]> que = new PriorityQueue<>((a, b) -> a[0] + a[1] - b[0] - b[1]);
3 List<List<Integer>> res = new ArrayList<>();
4 if (nums1.length == 0 || nums2.length == 0 || k == 0)
5 return res;
6 for (int i = 0; i < nums1.length && i < k; i++)
7 que.offer(new int[]{nums1[i], nums2[0], 0});
8 while (k-- > 0 && !que.isEmpty()) {
9 int[] cur = que.poll();
10 res.add(Arrays.asList(cur[0], cur[1]));
11 if (cur[2] == nums2.length - 1)
12 continue;
13 que.offer(new int[]{cur[0], nums2[cur[2] + 1], cur[2] + 1});
14 }
15 return res;
16}
解析：
题目说了nums1和nums2是排过序的，首先把nums1中的每个元素和nums2中的第一个元素进行组合，然后存放到队列中，队列que会根据每个组合的和的大小对他们进行排序，所以第一个最小的组合毫无疑问是[nums1[0]，nums2[0]]，在while循环中是第一个出队的，然后再把[nums1[0]，nums2[1]]放入到队列中（加入到队列中的组合都会进行排序的），然后这样进行不停的循环，直到找出k个为止。我们可以参照上面的数据画个图来分析一下
往期精彩试题
