给你一个包含 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
⽤ map 提前计算好任意 2 个数字之和,保存起来,可以将时间复杂度降到 O(n^2)。这⼀题⽐较麻烦的⼀点在于,最后输出解的时候,要求输出不重复的解。数组中同⼀个数字可能出现多次,同⼀个数字也可能使⽤多次,但是最后输出解的时候,不能重复。例如 [-1,-1,2] 和 [2, -1, -1]、[-1, 2, -1] 这 3 个解是重复的,即使 -1 可能出现 100 次,每次使⽤的 -1 的数组下标都是不同的。
复杂度分析
def threeSum(nums):
n=len(nums)
res=[]
if(not nums or n<3):
return res
nums.sort()
print(nums)
for i in range(n):
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=L+1
while(L<R and nums[R]==nums[R-1]):
R=R-1
L=L+1
R=R-1
elif(nums[i]+nums[L]+nums[R]>0):
R=R-1
else:
L=L+1
return res