python LeetCode Find K Pairs with Smallest Sums

最新推荐文章于 2026-01-09 15:10:18 发布

原创最新推荐文章于 2026-01-09 15:10:18 发布 · 556 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #python

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文介绍了一种算法，用于从两个整数列表中找到k个最小的数对组合。通过将所有可能的组合排序并选择前k个来实现，讨论了这种方法的时间复杂度，并暗示了使用堆数据结构可以更高效地解决此问题。

居然没超时
应该是要用堆来做

class Solution:
    def kSmallestPairs(self, nums1, nums2, k):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :type k: int
        :rtype: List[List[int]]
        """ 
        res=[]
        for n in nums1:
            for m in nums2:
                res.append([n,m])
        res1=sorted(res,key=lambda res:res[0]+res[1])
        return res1[:k]