【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]
]

思路：
排序+双指针，转化为 a+b 是否等于 -c的 问题
1.对数组nums排序后，遍历数组，每次固定位置i的元素，在[i+1,len(nums)-1]的范围内寻找两个和为c的数字。
2.使用left和right指针访问数组元素进行比较，直到两个指针相遇：

• 当nums[left]+nums[right]=-nums[i]时，把三个元素加入结果集；
• 当nums[left]+nums[right]<-nums[i]时，left指针右移；
• 当nums[left]+nums[right]>-nums[i]时，right指针左移；

3.为了避免访问到重复的元素，要对i,left,right将要指向的的下一个元素做判断，跳过相同元素。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res = []
        nums = sorted(nums)
        for i in range(len(nums)):
            if i>0 and nums[i]==nums[i-1]: continue
            left = i+1
            right = len(nums) - 1
            while left < right:
                if nums[left] + nums[right] == -nums[i]:
                    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
                    left+=1
                    right-=1
                elif nums[left] + nums[right] < -nums[i]: left+=1
                else: right-=1
        return res