给你一个包含 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]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
错误代码:
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
result=[]
result1=[]
if len(nums)<3:
return []
for i in range(len(nums)):
for j in range(i+1,len(nums)):
for k in range(j+1,len(nums)):
if nums[i]+nums[j]+nums[k] == 0:
result.append([nums[i],nums[j],nums[k]])
for i in result:
i.sort()
for i in result:
if i not in result1:
result1.append(i)
return result1
nums=[-1,0,1,2,-1,-4]
s=Solution()
print(s.threeSum(nums))
时间花费太长了,有更优解。
正确解法:
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
result=[]
for first in range(len(nums)):#遍历第一重循环
if first >0 and nums[first]==nums[first-1]:#确保不与之前的重复
continue
third = len(nums)-1#从最右端开始
target=-nums[first]
for second in range(first+1,len(nums)):#遍历第二重循环,并注意是从第一个已经遍历了
if second>first+1 and nums[second]==nums[second-1]:
continue#确保不重复
while second<third and nums[third]+nums[second]>target:
third-=1#比0大故减小
if second==third:
break#关键条件跳出
if nums[third]+nums[second]==target:
result.append([nums[first],nums[second],nums[third]])
return result
解析:首先进行排序,便于后续的第二重遍历和第三重遍历。采取双指针的方法,分别从左边与从右边向中间靠拢,直到出现第二个大于第三个跳出来即可。