LeetCode-15 三数之和（python）

最新推荐文章于 2022-06-07 21:41:05 发布

漂流瓶zps

最新推荐文章于 2022-06-07 21:41:05 发布

阅读量307

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签：三数之和 leetcode python

本文链接：https://blog.youkuaiyun.com/qq_33154865/article/details/87194308

leetcode 专栏收录该内容

23 篇文章

订阅专栏

该博客介绍了如何使用Python解决LeetCode中的第15题——三数之和。博主通过将问题转化为两数之和的问题，提出先对数组排序，然后采用前后夹逼的方法来寻找满足条件的三数之和，时间复杂度为O(n^2)。此方法同样适用于寻找k数之和，复杂度为O(max{n*logn, n^(k-1)})。博客中给出了详细的代码解析，并附有GitHub链接供读者参考。" 120022630,9295234,Ubuntu SSH连接自动断开：解决心跳超时问题,"['ssh', 'linux', 'ubuntu']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：三数之和

分析：写过“两数之和”可以稍作思考，找到a,b,c满足a+b+c=0，就是找到a,b满足a+b=-c，这就是两数之和了，再防止重复的结果。

这里给出相对通用的方案：先对数组进行排序，再前后夹逼求解，时间复杂度O(n^2)

不单单是三数之和，k数之和也可以这样推广，O(max { n*logn, n^(k-1) } )

直接看代码，在注释里解释，所得结果在数组中的下标为 i, j, k

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        target = 0
        result = []  # 我们要返回的结果集
        L_num = len(nums)
        if L_num < 3:
            return result  # 长度小于3，直接返回空结果
        num = sorted(nums)  # 先对数组进行排序，后续的操作都是为了避免重复的结果
        for i in range(L_num-2):
            if num[i] > 0: # 太牛批了，这个，理解一下，后面的值都不用考虑了
                break      # 
            j = i + 1
            if i > 0 and num[i] == num[i-1]:
                continue  # 这一步是为了排除重复的结果，后面的都是
            k = L_num - 1
            while j < k:
                if num[i] + num[j] + num[k] < target:
                    j = j + 1
                    while num[j] == num[j-1] and j < k:
                        j = j + 1
                elif num[i] + num[j] + num[k] > target:
                    k = k - 1
                    while num[k] == num[k+1] and j < k:
                        k = k - 1
                else:
                    print([num[i], num[j], num[k]])
                    result.append([num[i], num[j], num[k]])
                    j = j + 1
                    k = k - 1
                    while num[j] == num[j-1] and num[k] == num[k+1] and j < k:
                        j = j + 1
            # end while
        # end for
        return result

刷题GitHub：https://github.com/letterbeezps/leetcode-algorithm