三数之和，排序+双指针

"""
给定一个包含 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) :
        l = len(nums)
        nums.sort()  # 从小 --> 大
        res= []
        for i in range(0,l-1):
            if nums[i] > 0: break
            # 保证同一个数只固定一次，既相同数的最后一次出现
            if i > 0 and nums[i] == nums[i - 1]: continue
            left,right = i+1,l-1 # 定义左右指针
            while left < right:
                s = nums[i]+nums[left]+nums[right]
                if s==0:
                    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，right已经用过了，上面只是移动用过的left,right，在此基础上再+-
                    left+=1
                    right-=1
                elif s<0 :
                    while left<right and nums[left]==nums[left+1]:left+=1
                    left += 1
                else :
                    while left<right and nums[right]==nums[right-1]:right-=1
                    right-=1
        return res

测试：

if __name__ == "__main__":
    s = Solution()
    # 测试数据
    nums = [ -4, -1, -1, 0, 1, 2]
    nums = [-1,0,0,1]
    nums = [0,0,0,0]
    r = s.threeSum(nums)
    print(r)