三数之和

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

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

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

满足要求的三元组集合为：

[

[-1,0,1],

[-1,-1,2]

]

解题思路：先对序列排序，因为要找三个数，每次遍历固定一个元素，然后根据三数之和调整其他两个数的位置，如果等于0，符合要求，若果大于0，小的数字右移，如果小于0，大的数字左移。遍历过程中注意跳过重复数字。

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        #先对数组排序
        nums.sort()
        count = len(nums)
        collect = []
        for i in range(count):
            left = i+1
            right = count-1
            #避免重复找同一个数据
            if i >0 and nums[i] == nums[i-1]:
                left +=1
                continue
            #数据按小到大排列，每次选取nums[i]，在其后寻找符合a + b + c = 0的两个数据
            while left < right:
                sum = nums[i] + nums[left] + nums[right]
                if sum == 0:
                    col = [nums[i],nums[left],nums[right]]
                    collect.append(col)
                    left+=1
                    right-=1
                    #循环中nums[i]已确定，当再确定1个数时，另一个数也确定，左右端任一端出现重复均可跳过
                    while nums[left] == nums[left-1] and left < right:
                        left+=1
                    while nums[right] == nums[right+1] and left < right:
                        right-=1
                if sum<0:
                    #如果和小于0，小的数字右移增大
                    left+=1
                elif sum > 0:
                    #如果和大于0，大的数字左移减小
                    right-=1
        return collect

测试模块：

nums=[-1,0,1,2,-1,-4]
a=Solution.threeSum(1,nums)
print(a)

该算法还有待优化