给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
def threenum(nums):
nums.sort()
n=len(nums)
res=[]
temp=[]
for i in range(n):
if i>0 and nums[i]==nums[i-1]:
continue
left=i+1
right=n-1
while left<right:
cursum=nums[i]+nums[left]+nums[right]
if cursum==0:
temp.append(nums[i])
temp.append(nums[left])
temp.append(nums[right])
res.append(temp)
temp=[]
while left<right and nums[left]==nums[left+1]:
left+=1
while left<right and nums[right]==nums[right-1]:
right-=1
left+=1
right-=1
elif cursum>0:
right-=1
else:
left+=1
return res