题目描述
给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
示例
示例 1：
给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]
解答方法
方法一：排序 + 双指针
思路
- 首先对数组进行排序，排序后固定一个数nums[i]，再使用左右指针指向nums[i]后面的两端，数字分别为 nums[L]和 nums[R]，计算三个数的和sum 判断是否满足为 0，满足则添加进结果集
- 如果nums[i]大于 0，则三数之和必然无法等于 0，结束循环
- 如果 nums[i]==nums[i−1]，则说明该数字重复，会导致结果重复，所以应该跳过
- 当 sum == 0时，nums[L] == nums[L+1] 则会导致结果重复，应该跳过，L++
- 当 sum ==0 时，nums[R] == nums[R−1] 则会导致结果重复，应该跳过，R−−
代码
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
n = len(nums)
res =[]
if n<3:
return res
nums.sort()
for i in range(n-1):
if nums[i]>0:
return res
if i >0 and nums[i] == nums[i-1]:
continue
l = i + 1
r = n - 1
while l<r:
if nums[i] + nums[l] + nums[r] == 0:
res.append([nums[i], nums[l], nums[r]])
while l<r and nums[l] == nums[l+1]:
l+=1
while l<r and nums[r] == nums[r-1]:
r -= 1
l += 1
r -=1
elif nums[i] + nums[l] + nums[r] < 0:
l += 1
else:
r -= 1
return res
时间复杂度
O(n^2)，数组排序 O(NlogN)，遍历数组 O(n)，双指针遍历O(n)，总体 O(NlogN)+O(n)∗O(n)，O(n^2 )
空间复杂度
O(1)
