原题地址39. 组合总和
给定一个无重复元素的数组 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]]
算法
迭代,先对candidates从小到大排序,每次用target减去candidates[i],如果结果刚好为0,就输出;如果结果小于0,说明已经溢出了,结束;如果结果大于0,那么将candidates[i]记录进结果,并更新target,以及candidates更改为不小于candidates[i]的部分,保证选出的结果不重复。
代码
class Solution:
def combinationSum(self, candidates, target):
"""
:type candidates: List[int]
:type target: int
:rtype: List[List[int]]
"""
candidates.sort()
def back(lst, tg, res):
for i in range(len(lst)):
if tg - lst[i] == 0:
yield res + [lst[i]]
if tg - lst[i] > 0:
yield from back(lst[i:], tg - lst[i], res + [lst[i]])
if tg - lst[i] < 0:
break
return list(back(candidates, target, []))