[LeetCode]40、组合总和2
2019-07-28 本文已影响0人
河海中最菜
题目描述
给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。
说明:
所有数字(包括目标数)都是正整数。
解集不能包含重复的组合。
示例 1:
输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]
思路解析
只能用一次,就是不要继续重复使用当前元素,在上一题的基础加了一点的改动。
class Solution:
def combinationSum2(self, candidates, target):
res = []
if not candidates or len(candidates) == 0:
return res
candidates.sort()
def helper(temp, temp_sum, i):
if temp_sum == target:
if temp not in res:
res.append(temp)
return
if temp_sum > target or i >= len(candidates):
return
helper(temp + [candidates[i]], temp_sum + candidates[i], i + 1)
helper(temp, temp_sum, i + 1)
helper([], 0, 0)
return res
def combinationSum2_(self, candidates, target):
if not candidates or len(candidates) == 0:
return []
candidates.sort()
res = []
def helper(temp, temp_sum, index):
if temp_sum == target:
res.append(temp)
return
for i in range(index, len(candidates)):
# 判断当前元素是否能够加入组合
if temp_sum + candidates[i] > target:
break
# 避免重复元素,并不是只一个元素能够重复多次的那种
if i > index and candidates[i] == candidates[i-1]:
continue
helper(temp + [candidates[i]], temp_sum + candidates[i], i+1)
helper([], 0, 0)
return res
AC40