15. 三数之和

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

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

示例：

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

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

思路：双指针法

先把nums从小到大排序。

先固定a, a是nums[0], 然后，left指针指向a右边的第一个元素，right指针指向nums末尾的元素。

如果a+nums[left]+nums[right]<0, 说明left指向的元素小了，把left指针往右边移动一个位置。

如果a+nums[left]+nums[right]>0, 说明right指向的元素大了，把right指针往左边移动一个位置。

重复上述操作，比如，如果把left往右移动了一个位置之后，变成了三数之和大于0，那就把right指针往左边移动一个位置。

直到三数之和等于0，我们就找到了。如果始终不等于0，那就说明这个a没有三数之和等于0。

把a往右移动一个位置，然后重复上述操作。

注意：要做去重：因为题目要求答案中不能有重复的三元素。

所以，a=nums[i], left, right都要做去重。

【注意】判断是否相同元素的时候，不能判断nums[i]是否等于nums[i+1]，因为，结果集里面是可以有重复的元素的。比如[-1,-1,2]。如果你要判断nums[i]是否等于nums[i+1]，相当于让结果集里不能有重复的元素，这是不对的。

应该判断nums[i]是否等于nums[i-1]

还有，对left和right也要进行去重。

判断nums[left]是否等于nums[left-1], nums[right]是否等于nums[right+1]

去重要放在收获结果的下面，至少收获了一个结果，再判断是否重复。否则，可能一直重复，一直去掉重复，直到left==right，那么一个结果也没收获，是不对的。

注意：如果排序后，第一个元素就>0，那就不需要进一步检查了，肯定没有三个元素相加=0.

我写的：

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res = list()
        nums = sorted(nums)
        for i in range(len(nums)):
            if nums[i] > 0:
                return res

            left = i+1
            right = len(nums) - 1

            while left < right:
                sum = nums[i] + nums[left] + nums[right]
                if sum < 0:
                    left += 1
                elif sum > 0:
                    right -= 1
                else:
                    res.append([nums[i], nums[left], nums[right]])
                    if nums[left] == nums[left+1]:
                        left += 1
                    elif nums[right] == nums[right-1]:
                        right -= 1

        return res

超过运行时间了。

nums = sorted(nums) 可以写成nums.sort( )

在判断完nums[i]>0之后，判断是否重复：
 

if i>0 and nums[i] == nums[i-1]:

continue

【注意：一定要加上条件i>0, 因为如果i=0，index=i-1是list最后一个元素，这是不对的。】

最后判断left和right是否有重复，用while循环，而不仅仅是if statement：

把：
 

if nums[left] == nums[left+1]:

left += 1

elif nums[right] == nums[right-1]:

right -= 1

改成：

while left < right and nums[left] == nums[left+1]:

left += 1

while left < right and nums[right] == nums[right-1]:

right -= 1

（记住，要时刻加上判断条件 left<right）

最后，right -= 1

left += 1

【注意：找到一组和为0的三个数之后，还要移动left和right，继续找！】

修改后，正确版：

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res = list()
        nums.sort()

        for i in range(len(nums)):
            if nums[i] > 0:
                return res

            if i>0 and nums[i] == nums[i-1]:
                continue

            left = i+1
            right = len(nums) - 1

            while left < right:
                sum = nums[i] + nums[left] + nums[right]
                if sum < 0:
                    left += 1
                elif sum > 0:
                    right -= 1
                else:
                    res.append([nums[i], nums[left], nums[right]])

                    while left < right and nums[left] == nums[left+1]:
                        left += 1
                    while left < right and nums[right] == nums[right-1]:
                        right -= 1
                
                right -= 1
                left += 1
        return res