【边学边敲边记】LeetCode006:三数之和

一、写在前面

上一题传输门：
今天给大家分享的是LeetCode 数组与字符串第六题：三数之和，为面试而生，期待你的加入。

二、今日题目

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：
答案中不可以包含重复的三元组。

示例:

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

三、分析

看到这个题目第一眼，我想到的就是两数之和，咋一眼感觉上没有太大区别，实际是二维空间和三维空间的区别，最开始想的是和两数之和方法类似，这里我只要找到两数之和的相反数是不是在列表就行了，但真的实践起来，比较麻烦，时间复杂度先不说，就是思想上随便一说都是个麻烦事情，所以X先生果断放弃了这样掉头发不讨好的方法，换了一种思想，头+尾搜索法：头部循环，尾部从两端向中间搜索，思想如下：

【边学边敲边记】LeetCode006:三数之和

我的思想

四、解题

我的方法：
从从中心向外扩散，时间复杂度：小于O(n^2)

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        # 列表排序，从小到大
        nums.sort()
        # [-4, -1, -1, 0, 1, 2]
        res_list = []
        # 头部循环查找
        for i in range(len(nums)):
            if i == 0 or nums[i] > nums[i - 1]:
                # 最左端
                l = i + 1
                # 最右端
                r = len(nums) - 1
                while l < r:  # 正在查找
                    three_sum = nums[i] + nums[l] + nums[r]
                    if three_sum == 0:
                        res_list.append([nums[i], nums[l], nums[r]])
                        l += 1 # 右移一位
                        r -= 1 # 左移一位
                        while l < r and nums[l] == nums[l - 1]:
                            # 从左往右，相同数值直接跳过
                            l += 1
                        while r > l and nums[r] == nums[r + 1]:
                            # 从右往左，相同数值直接跳过
                            r -= 1
                    elif three_sum > 0:
                        # 大于零，右边数值大，左移
                        r -= 1
                    else:
                        # 小于零，左边数值小，右移
                        l += 1
        return res_list

提交结果

【边学边敲边记】LeetCode006:三数之和

提交结果

测试数据：313组
运行时间：740ms
击败人百分比:81.35%

五、疑惑

这里说一下上一篇大牛方法中的zip()。

zip() 函数用于将可迭代的对象作为参数，将对象中对应的元素打包成一个个元组，然后返回由这些元组组成的列表。

如果各个迭代器的元素个数不一致，则返回列表长度与最短的对象相同，利用 * 号操作符，可以将元组解压为列表。

zip 方法在 Python 2 和 Python 3 中的不同：在 Python 3.x 中为了减少内存，zip() 返回的是一个对象。如需展示列表，
需手动 list() 转换。