leetcode python3 三数之和

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

如果使用暴力法，时间复杂度为O(N3)，超出时间限制，因此使用双指针法解决该问题,首先需要对原始数组进行排序，其时间复杂度（O(nlogn））,排序后的数组有序以便解决重复三元组问题，基础思路：固定最左边一个数组，对其右边部分使用双指针法处理。
双指针法思路： 固定指针中最左（最小）数字的指针 k，双指针 i，j 分设在数组索引 (k, len(nums))(k,len(nums)) 两端，通过双指针交替向中间移动，记录对于每个固定指针 k 的所有满足 nums[k] + nums[i] + nums[j] == 0 的 i,j 组合：
当 nums[k] > 0 时直接break跳出：因为 nums[j] >= nums[i] >= nums[k] > 0，即 3个数字都大于 0 ，在此固定指针 k 之后不可能再找到结果了。
当 k > 0且nums[k] == nums[k - 1]时即跳过此元素nums[k]：因为已经将 nums[k - 1] 的所有组合加入到结果中，本次双指针搜索只会得到重复组合，k>0的关键是处理【0，0，0】的特例，如不加，则会输出【】。
i，j 分设在数组索引 (k, len(nums))(k,len(nums)) 两端，当i < j时循环计算s = nums[k] + nums[i] + nums[j]，并按照以下规则执行双指针移动：
当s < 0时，i += 1并跳过所有重复的nums[i]；
当s > 0时，j -= 1并跳过所有重复的nums[j]；
当s == 0时，记录组合[k, i, j]至res，执行i += 1和j -= 1并跳过所有重复的nums[i]和nums[j]，防止记录到重复组合。
复杂度分析：
时间复杂度 O(N^2)O(N 
2
 )：其中固定指针k循环复杂度 O(N)O(N)，双指针 i，j 复杂度 O(N)O(N)。
空间复杂度 O(1)O(1)：指针使用常数大小的额外空间。

class Solution:

    def threeSum(self, nums: List[int]) -> List[List[int]]:

        result = []

        nums.sort()

        if len(nums) < 3:

            return []

        k = 0

        for k in range(len(nums) - 2):

            if nums[k] > 0:

                break # 结束循环

            if k > 0 and nums[k] == nums[k-1]:

                continue # 跳出本次循环

            start = k + 1

            end = len(nums) - 1

            while start < end:

                res = nums[k] + nums[start] + nums[end]

                if res < 0:

                    start += 1

                    while start < end and nums[start] == nums[start - 1]:

                        start += 1  # 如果重复，继续加1

                elif res > 0:

                    end -= 1

                    while start < end and nums[end] == nums[end+1]:

                        end -= 1

                else:

                    result.append((nums[k], nums[start], nums[end]))

                    start += 1

                    end -= 1

                    while start < end and nums[start] == nums[start - 1]:

                        start += 1

                    while start < end and nums[end] == nums[end+1]:

                        end -= 1

        return result

核心思路：对数组排序以避免重复三元组，固定一个值，找另外二个值它们和等于 0，使用双指针法寻找这两个值。