三数之和——python

问题描述

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

暴力求解时间复杂度为O(n³)无法通过测试
感谢吴彦祖的题解：原来还可以这样

排序加双指针法

首先将列表有序化方便排除重复解和确定三个元素的相对位置
i为最小的数，l为中间数，r为最大数的下标
当nums[i] + nums[l] + nums[r] == 0时将其加入res中
注意排除三个数的相等值

代码

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