题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
代码(排序+双指针):
class Solution:
def threeSum(self, nums):
if len(nums) < 3: # 若数组长度 < 3,直接返回空
return []
nums = sorted(nums) # 排序
re = [] # 结果变量
for i in range(len(nums)): # 遍历
if nums[i] > 0: # 若 nums[i] > 0,则没有数字之和为0,直接返回re
return re
if i > 0 and nums[i] == nums[i-1]: # 排除重复数组
continue
L = i + 1 # 左指针
R = len(nums) - 1 # 右指针
while L < R: # 遍历
temp = nums[i] + nums[L] + nums[R] # 三数之和
if temp == 0: # 若temp = 0
re.append([nums[i],nums[L],nums[R]]) # 添加到结果列表中
while L < R and nums[L] == nums[L+1]: # 避免左指针下一个数字重复,跳过
L = L + 1
while L < R and nums[R] == nums[R-1]: # 避免右指针下一个数字重复,跳过
R = R - 1
L = L + 1 # 左指针右移,遍历
R = R - 1 # 右指针左移,遍历
elif temp > 0: # 若temp > 0
R = R - 1 # 则应该让三数之和小一些,故右指针左移
else:
L = L + 1 # 反之,应该让三数之和大一些,故左指针右移
return re