[Leetcode]组合总和
题目描述
给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。
说明:
所有数字(包括 target)都是正整数。
解集不能包含重复的组合。
示例 1:
输入:candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]
示例 2:
输入:candidates = [2,3,5], target = 8,
所求解集为:
[
[2,2,2,2],
[2,3,3],
[3,5]
]
提示:
1 <= candidates.length <= 30
1 <= candidates[i] <= 200
candidate 中的每个元素都是独一无二的。
1 <= target <= 500
解题思路
- 递归回溯
- 递归终止条件:
- 超出数组范围
- 当前所选的结果和大于target
- 当前所选的结果和等于target
- 递归调用:
- 选择当前数进行下一轮递归
- 跳过当前数进行下一轮递归
实现代码
class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
res=[]
def helper(i,cur_sum,ans):
if i == len(candidates) or cur_sum > target:
return
if cur_sum ==target:
res.append(ans)
return
helper(i, cur_sum + candidates[i], ans + [candidates[i]]) #选择当前数
helper(i + 1, cur_sum , ans) #跳过当前数
helper(0, 0, [])
return res
- 方法二
class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
res=[]
def helper(i,cur_sum,ans):
if i == len(candidates) or cur_sum > target:
return
if cur_sum ==target:
res.append(ans)
return
for j in range(i, len(candidates)): //类似于迭代的调用递归
helper(j, cur_sum + candidates[j], ans + [candidates[j]])
helper(0, 0, [])
return res
- 利用排序
class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
candidates.sort() //排序
res=[]
def helper(i,cur_sum,ans):
if i == len(candidates) or cur_sum > target:
return
if cur_sum ==target:
res.append(ans)
return
for j in range(i, len(candidates)):
if cur_sum + candidates[j] >target: //排序之后,如果已经大于target就不用继续递归后面
break
helper(j, cur_sum + candidates[j], ans + [candidates[j]])
helper(0, 0, [])
return res