leetCode.15.三数之和（中等）

题目描述：

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

输入输出实例：

思路：刚开始看到这道题目，我首先想的是直接暴力遍历就好了，找到符合条件的就放入结果列表，对结果列表里面的元素进行删重复操作，后面也不出意外的没有通过，确实超时了（悲）

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        n = len(nums)
        ans1 = []
        judge = set()
        for i in range(0,n-2):
            for j in range(i+1,n-1):
                for k in range(j+1,n):
                    ans2 = []
                    if nums[i] + nums[j] + nums[k] == 0:
                        sorted_set = tuple(sorted([nums[i],nums[j],nums[k]]))
                        if sorted_set not in judge :
                            judge.add(sorted_set)
                            ans2 = [nums[i],nums[j],nums[k]]
                            ans1.append(ans2)
        return ans1

然后我对这个方法进行了改进，采用了一个双指针的思路，就是首先数组进行排序 ，然后nums[i]作为三元组的第一个元素，在他的后面我们设置left和right指针，然后通过移动left和right指针来实现结果，当 left < right 时，进行循环：

• 计算当前三个数的和 total = nums[i] + nums[left] + nums[right]。
• 如果 total == 0，则找到了一个符合条件的三元组，将其加入结果列表。
• 如果 total < 0，说明当前的和小于0，需要增大和，因此将 left 指针向右移动。
• 如果 total > 0，说明当前的和大于0，需要减小和，因此将 right 指针向左移动。

根据上述思路，得到的代码如下：

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

 需要注意的是这段代码里left和right的移动的时候就已经帮助我们去除了重复的可能