Leetcode：三数之和

给你一个包含 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: List[int]) -> List[List[int]]:
        n = len(nums)
        nums.sort()
        ans = list()
        # 剪枝
        if n < 3 :return ans
        for i in range(n-2):
        #去除重复
            if i > 0 and nums[i] == nums[i-1]:continue
        #剪枝:因为是排序之后的，如果nums[i] + nums[i+1] + nums[i+2] > 0说明后面的也全会大于0，跳出循环，循环结束。nums[i] + sum(nums[-1:-3:-1])<0的话，当前循环结束，进入下一个循环。
        #这里的剪枝是关键，可以极大的排除很多因素
            if nums[i] + nums[i+1] + nums[i+2] > 0 :break
            if nums[i] + sum(nums[-1:-3:-1])<0:continue
        #双指针
            l, r = i+1, n-1
            while l < r:
                tmp = nums[i] + nums[l] + nums[r]
                if  tmp == 0:
                    ans.append([nums[i], nums[l], nums[r]])
                    r -= 1 
                    l += 1
                 	#这里是去除l和r位置重复的元素
                    while l < r and nums[r] == nums[r+1]:r -= 1
                    while l < r and nums[l] == nums[l-1]:l += 1
                elif tmp > 0 : 
                    r -= 1
                    continue
                else:
                    l += 1
                    continue
        return ans

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。