leetcode-15. 三数之和

15. 三数之和
给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

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

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
 

排序后双指针 [py]

class Solution:
    
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        if len(nums) < 3:
            return []

        nums.sort()
        res = []

        for i in range(0, len(nums)):

            if i > 0 and nums[i] == nums[i-1]:
                continue

            j = i + 1
            k = len(nums) - 1

            while j < k:
                
                curr = nums[i] + nums[j] + nums[k]
                moveJ = False
                moveK = False

                if curr == 0:
                    res.append([nums[i], nums[j], nums[k]])
                    moveJ = True
                    moveK = True

                elif curr > 0:
                    moveK = True
                else:
                    moveJ = True
                    
                if moveJ:
                    jv = nums[j]
                    # 移动时跳过相同的数 不然三元组会重复
                    while j < k and nums[j] == jv:
                        j += 1
                if moveK:
                    kv = nums[k]
                    while k > j and nums[k] == kv:
                        k -= 1

        return res