给定两个以升序排列的整数数组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]
提示:
1 <= nums1.length, nums2.length <= 10^4
-10^9 <= nums1[i], nums2[i] <= 10^9
nums1
,nums2
均为升序排列1 <= k <= 1000
1.暴力超时
Python解答:res = [[nums1[i], nums2[j]] for i in range(len(nums1)) for j in range(len(nums2))] res.sort(key=(lambda x:x[0]+x[1])) return res[:k]
留言