【题目】
Given an array nums
of n integers, are there elements a, b, c in nums
such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.
【注意】
The solution set must not contain duplicate triplets.
【举例】
Given array nums = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]
【题目翻译】
大意就是从一个数组里面找出所有和为0的三元组, 不能包括重复的, 举例如上。
【解题思路】
这题乍看可以用回溯法解决,
先选前三个数[-1,0,1]判断, 然后把第三个数去掉换成第四个数[-1,0,2]再判断, 然后把第四个数去掉换成第五个数[-1,0,-1]再判断...
判断完-1, 0 为头两个位置的数后, 再判断-1, 1为头两个位置的情况, 再判单-1, 2为头两个位置的情况...
判断为-1为起始位置的所有情况以后, 再判断0为起始位置的所有情况...
【回溯代码】
class Solution:
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
result = []
self.backtracking(result, [], 0, nums)
return result
def backtracking(self, result, tmp, start, nums):
if len(tmp) == 3:
if sum(tmp) == 0:
result.append(list(tmp))
return
for i in range(start, len(nums)):
if len(tmp) < 3:
tmp.append(nums[i])
self.backtracking(result, tmp, i+1, nums)
tmp.pop()
输出结果是:
[[-1,0,1],[-1,2,-1],[0,1,-1]]
发现回溯法解决不了数组重复的问题。
其实这道题还有另一种解法:
首先对数组nums进行升序排序, 然后遍历所有三元组可能的起始元素nums[0], nums[len(nums)-3], 假设遍历到了起始元素nusm[i], 然后再剩余元素里面找两个和为-nums[i]的元素, 与起始元素组成和为零的三元组。
如果起始元素nums[i] == nums[i-1], nums[i]为起始元素的可能就不用再考虑, 因为所有可能已经包含在了nums[i-1]为起始元素的情况里。
【代码】
class Solution(object):
def threeSum(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
nums.sort()
res = []
for i in range(len(nums)-2):
if i == 0 or nums[i] != nums[i-1]: #如果该起始元素等于上一个起始元素,跳过
#在剩余元素里找和为-nums[i]的组合
lo = i + 1
hi = len(nums) - 1
sum_ = -nums[i]
while lo < hi:
if nums[lo] + nums[hi] == sum_:
res.append([nums[i], nums[lo], nums[hi]])
while lo < hi and nums[lo+1] == nums[lo]: lo += 1 #避免重复
while lo < hi and nums[hi-1] == nums[hi]: hi -= 1
hi -= 1
lo += 1
elif nums[lo] + nums[hi] > sum_:
hi -= 1
else:
lo += 1
return res