给定一个无重复元素的数组 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]
]
一开始,博主也尝试过分类讨论,发现当出现target可以由列表中的多个元素、若干元素重复+不重复元素的分类时,会十分复杂。最终还是选择使用DFS实现。
class Solution(object):
def combinationSum(self, candidates, target):
"""
:type candidates: List[int]
:type target: int
:rtype: List[List[int]]
"""
candidates.sort()
self.result = []
self.DFS(candidates, target, 0, [])
return self.result
def DFS(self, candidates, target, start, marklist):
if target == 0:
return self.result.append(marklist)
for i in range(start, len(candidates)):
if candidates[i] > target:
return
self.DFS(candidates, target - candidates[i], i, marklist+[candidates[i]])