LeetCode-查找和最小的K对数字---多种解法

题目：查找和最小的K对数字

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

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

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

例 :

输入: 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]

来源：力扣（LeetCode）
链接：
以下python高级用法参见：https://blog.csdn.net/qq_19446965/article/details/104070046

1、暴力法

枚举所有组合，按照和排序，最后返回前 K 项。

import itertools
def kSmallestPairs(self, nums1, nums2, k):
    # sorted(itertools.product(nums1, nums2), key=sum)[:k]  # 生成的是元组不是列表
    return map(list, sorted(itertools.product(nums1, nums2), key=sum)[:k])

2、堆

仍然遍历所有对，但仅使用生成器和使用大小为 k 的堆heapq.nsmallest。 因此，这仅需要O(k)O(k)的额外内存和O(mn log k)O(mnlogk)的时间。

import heapq
def kSmallestPairs(self, nums1, nums2, k):
    return map(list, heapq.nsmallest(k, itertools.product(nums1, nums2), key=sum))

不计算整个矩阵，而将每一行变成三元组[u + v，u，v]的生成器，仅在要求计算下一行时才计算下一行，然后将这些生成器与堆合并。 占用 O(m + k * log(m))O(m+k∗log(m)) 的时间复杂度和 O(m)O(m) 的空间复杂度。

import heapq
def kSmallestPairs(self, nums1, nums2, k):
    streams = map(lambda u: ([u+v, u, v] for v in nums2), nums1)
    stream = heapq.merge(*streams)
    return [suv[1:] for suv in itertools.islice(stream, k)]

3、堆+BFS

从矩阵左上角的第一对开始，广度优先遍历（右+下）。 每当将一对选择为输出结果时，该行中的下一对就会添加到当前选项的堆中。

BFS模板参见：https://blog.csdn.net/qq_19446965/article/details/102739705

def kSmallestPairs(nums1, nums2, k):
    queue = []
    if nums1 and nums2:
        heapq.heappush(queue, [nums1[0] + nums2[0], 0, 0])
    res = []
    visited = {(0, 0)}
    while queue and len(res) < k:
        _, i, j = heapq.heappop(queue)
        res.append([nums1[i], nums2[j]])
        if j+1 < len(nums2) and (i, j+1) not in visited:
            heapq.heappush(queue, [nums1[i] + nums2[j+1], i, j+1])
            visited.add((i, j+1))
        if i+1 < len(nums1) and (i+1, j) not in visited:
            heapq.heappush(queue, [nums1[i+1] + nums2[j], i+1, j])
            visited.add((i+1, j))
    return res

——————————————————————

参考：https://leetcode-cn.com/problems/find-k-pairs-with-smallest-sums/solution/pyhont3yi-xing-jie-fa-dao-gao-xiao-jie-fa-by-ml-zi/ （LeetCode）