给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:先对数组进行排序,然后使用双指针,左右逼近,遍历数组作为外层循环,重点是去重,left或right碰到相同数字则往下跳一个
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
#-4,-1,-1,0,1,2
#第一步,判断边界
res = []
nums.sort()
n=len(nums)
for i in range(n-2):
if i==0 or nums[i]>nums[i-1]:#去重
left,right = i+1,n-1
while left<right:
ident = nums[i]+nums[left]+nums[right]
if ident==0:
res.append([nums[i],nums[left],nums[right]])
left+=1
right-=1
#去重
while left<right and nums[left]==nums[left-1]:
left+=1
while left<right and nums[right]==nums[right+1]:
right-=1
elif ident<0:
left+=1
else:
right-=1
return res
原文:https://www.cnblogs.com/xiao-longxia/p/12496290.html