candidates 中的数字可以无限制重复被选取。
class Solution:
def combinationSum(self, candidates, target):
# 结果列表
ans = []
# 可能的组合
tmp = []
def helper(idx, total):
"""回溯,求组合总和
Args:
idx: 选取元素索引
total: 组合中的元素和
"""
# 基准条件
# 当元素和大于目标值,直接返回
if total > target:
return
# 当元素和等于目标值,将组合添加到结果中,返回
if total == target:
ans.append(tmp[::])
return
# 进入分支,同时避免重复组合
for i in range(idx, len(candidates)):
# 更新 total 值,
total += candidates[i]
# 同时将当前元素尝试添加到组合中
tmp.append(candidates[i])
# 再次进入递归
# 这里可以看文章图例,递归向下,可选元素是从自身开始选择
# 这里同时也能避免组合重复,因为不会再次选择索引 i 前面对应的元素
helper(i, total)
# 回溯,回退组合元素及 total 值
tmp.pop()
total -= candidates[i]
total = 0
helper(0, total)
return ans