15.三数之和

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]]:
        nums.sort()
        ans = []
        for i in range(len(nums) - 2):
            if i > 0 and nums[i] == nums[i-1]: # 跳过重复的第一个数子
                continue
            j = i + 1
            k = len(nums) - 1
            while j < k:
                sum_ = nums[i] + nums[j] + nums[k]
                if sum_ == 0:
                    ans.append([nums[i], nums[j], nums[k]])
                if sum_ > 0:
                    flag = 1
                    # 跳过重复的第三个数子
                    while k - flag > i + 1 and nums[k] == nums[k - flag]:
                        flag += 1
                    k -= flag
                else:
                    flag = 1
                    # 跳过重复的第二个数子
                    while j + flag < len(nums) and nums[j] == nums[j + flag]:
                        flag += 1
                    j += flag
        return ans

思路

使用的官方题解。利用加和固定值，三数字中一个变大一个不变第三个数一定变小的原理，将三层循环优化为一层循环加一层双指针循环。

使用python列表自带排序函数进行升序排序，这里注意nums.sort()是对nums列表本身的内存空间数据进行排序，返回值是空。排完序后进行枚举结果。

枚举过程中注意一个点，因为要求答案是不重复的，所以在i, j, k的递增递减上要进行一个重复值跳过的操作。