给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
classSolution(object):defthreeSum(self, nums):"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
new =[]
k =0
t =len(nums)-2for k inrange(t):if nums[k]>0:breakif k >0and nums[k]==nums[k-1]:continue
i , j = k+1,len(nums)-1while i < j:
s = nums[k]+nums[i]+nums[j]if s <0:
i +=1while i < j and nums[i]== nums[i -1]:
i +=1elif s >0:
j -=1while i < j and nums[j]==nums[j +1]:
j -=1else:
new.append([nums[k],nums[i],nums[j]])
i +=1
j -=1while i < j and nums[i]== nums[i -1]:
i +=1while i < j and nums[j]== nums[j +1]:
j -=1return new