leetcode-三数之和python

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

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

示例：

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

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

思路：先对数组进行排序，然后使用双指针，左右逼近，遍历数组作为外层循环，重点是去重，left或right碰到相同数字则往下跳一个

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        #-4,-1,-1,0,1,2
        #第一步，判断边界
        res = []
        nums.sort()
        n=len(nums)
        for i in range(n-2):
            if i==0 or nums[i]>nums[i-1]:#去重    
                left,right = i+1,n-1
                while left<right:
                    ident = nums[i]+nums[left]+nums[right]
                    if ident==0:
                        res.append([nums[i],nums[left],nums[right]])
                        left+=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
                    elif ident<0:
                        left+=1
                    else:
                        right-=1
        return res