LeetCode 39. 组合总和
39. 组合总和
Difficulty: 中等
给定一个无重复元素的数组 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
Solution
很基础的一道考察组合题目,一般排列组合类型的问题考察的就是回溯算法(组合总和IV那道题比较特殊,考察的是动态规划,使用回溯算法会超时)。有很多大佬总结过回溯算法经典解题框架(套路),简单来说就是:
result = []
def backtrack(路径, 选择列表):
if 满足结束条件:
result.add(路径)
return
for 选择 in 选择列表:
做选择
backtrack(路径, 选择列表)
撤销选择
用这个框架来解决问题的时候,需要注意两个问题:1. 递归什么时候结束?2. 元素是可以被重复使用的还是不可以被重复使用的?
回到这个题目,题目给定了无重复元素的数组,并且数组内的元素可以被重复使用,只要保证从数组内选择的元素求的和等于target
就行了。好,现在我们考虑上文提到需要考虑的两个问题:1. 数组内元素每次被使用之后,target
减去该元素,直到target
被减到0,说明找到了满足和为target
的组合。2. 题目交代数组内的元素可以被重复使用,那么选择的列表就是candidates
。回答了这两个问题,剩下的代码就好办了。
解法一:
class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
path = []
res = []
self.backtrack(candidates, path, target, res)
return res
def backtrack(self, nums, path, target, res):
if target < 0:
return
if target == 0:
res.append(path[:])
return
for i in range(len(nums)):
path.append(nums[i])
self.backtrack(nums, path, target-nums[i], res)
path.pop()
解法二:
class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
res = []
self.dfs(candidates, target, [], res)
return res
def dfs(self, nums, target, path, res):
if target < 0:
return
if target == 0:
res.append(path)
return
for i in range(len(nums)):
self.dfs(nums[i:], target-nums[i], path+[nums[i]], res)