Leetcode:15 三数之和

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

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

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

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

 tips:三个数之和暴力搜索时间复杂度为n**3。1.我们可以先固定一个数，将其转化为两数之和。那么使用双指针一次便利求出两数之和需要数组有序，故先排序。2.其次，排序后的数从小到大，从负到正，若第一个数为正数，那三数之和不可能为0，故剪枝。3.若要去掉重复项，我们需要判断：固定的第一个数在此之前是否存在，若存在则跳过；第二个数的右侧与第三个数的左侧是否存在相同的值，若存在，跳过。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        res = []
        if len(nums)<3:
            return res
        for num in range(len(nums)):
            if nums[num]>0:
                return res
            if num>0 and nums[num]==nums[num-1]:
                continue
            p1=num+1
            p2=len(nums)-1
            while p1<p2:
                if nums[p1]+nums[p2]+nums[num]==0:
                    # if [nums[num],nums[p1],nums[p2]] not in res:
                    res.append([nums[num],nums[p1],nums[p2]])
                    while p1<p2 and nums[p1]==nums[p1+1]:
                        p1+=1
                    while p1<p2 and nums[p2]==nums[p2-1]:
                        p2-=1
                    p1+=1
                    p2-=1
                elif nums[p1]+nums[p2]+nums[num]>0:
                    p2-=1
                else:
                    p1+=1
        return res