leetcode-15. 三数之和python3

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

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

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

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

class Solution:
    def threeSum(self, nums):
        '''

        思路: a,b,c三个数除了[0,0,0]这种组合，其中一定需要有负数。所以有以下过程:
            1.排序
            2.循环list遍历a,如果a>0 终止循环
            3.a的右面是一个新的list，在新的list种b在左端c在右端，如果abc之和>0 c向左移动，如果abc之和小于0b向右移动
        '''
        nums = sorted(nums)
        print(nums)
        res = []
        
        for i in range(len(nums) - 2):
            if i>0 and nums[i]==nums[i-1]:continue
            if nums[i]>0:break
            left,right = i+1,len(nums)-1

            while left<right:
                a = nums[i]
                b = nums[left]
                c = nums[right]
                sums = a+b+c

                if sums>0:right-=1

                elif sums<0:left+=1
                else:
                    res.append([a,b,c])
                    left+=1
                    right-=1
                    while nums[left]==nums[left-1] and left<right:left+=1
                    while nums[right] == nums[right+1] and left < right: right -= 1

        return res