15. 三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意: 答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[ [-1, 0, 1], [-1, -1, 2] ]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解
先回忆一下两数之和,比较一下,我们可以发现,在求三数之和时,当我们确定了第一个数,问题就变成了求两数之和。所以这个问题可以等价于两数之和外套一个循环固定三数之和的第一个数。第一步都是排序。有由于两数之和有双指针和哈希两种解法,三数之和也可以有两种解法
这是我第一次根据两数之和改写来的,但是会出现重复解的现象
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
n=len(nums)
if(not nums or n<3):
return []
nums.sort()
res=[]
for i in range(n):
if(nums[i]>0):
return res
if(i>0 and nums[i]==nums[i-1]):
continue
cur = 0 - nums[i]
dict = {}
for j, item in enumerate(nums):
if dict.get(cur - item) is not None:
res.append([nums[i],nums[dict.get(cur - item)],nums[j]])
else:
dict[item] = j
return res
在此基础上修复错误
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
n=len(nums)
res=[]
if(not nums or n<3):
return res
nums.sort()
dic1={}
#dic1用来去掉固定的第一位重复的情况,已及解决[0,0,0]
#如果下面range(n-2)用的是range(n),也可以不用字典实现这一步骤
for i in range(n-2):
if nums[i]>0:return res
if nums[i] in dic1:
continue
else: dic1[nums[i]]=True
target = - nums[i]
cur_nums = nums[i+1:]
dic2 = {}
for j in range(len(cur_nums)):
print(nums[i] ,cur_nums[j],dic2)
if (target - cur_nums[j]) in dic2:
if dic2[target - cur_nums[j]] == False:
res.append([nums[i],target - cur_nums[j],cur_nums[j]])
dic2[target - cur_nums[j]] = True
else:
continue
else:
dic2[cur_nums[j]] = False
return res
字典的好处是代码简介,但是不好理解,因为没有对排序后的顺序太好的利用,而双指针就严格依据数学原理,每一步都显得有序,更好理解,对时间空间的利用率更高,更完美。
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
result = []
nums.sort()
for i in range(len(nums)):
if i > 0 and nums[i] == nums[i - 1]:
continue
left, right = i + 1, len(nums) - 1
while left < right:
cur_Sum = nums[i] + nums[left] + nums[right]
if cur_Sum == 0:
result.append([nums[i], nums[left], nums[right]])
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 cur_Sum < 0:
left += 1
else:
right -= 1
return result