给定一个无重复元素的正整数数组 candidates 和一个正整数 target ,找出 candidates 中所有可以使数字和为目标数 target 的唯一组合。
candidates 中的数字可以无限制重复被选取。如果至少一个所选数字数量不同,则两种组合是唯一的。
对于给定的输入,保证和为 target 的唯一组合数少于 150 个。
示例 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]]
示例 3:
输入: candidates = [2], target = 1
输出:
示例 4:
输入: candidates = [1], target = 1
输出: [[1]]
示例 5:
输入: candidates = [1], target = 2
输出: [[1,1]]
提示:
1 <= candidates.length <= 30
1 <= candidates[i] <= 200
candidate 中的每个元素都是独一无二的。
1 <= target <= 500
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一解
先排序,然后使用深度优先遍历求解,在求解中进行剪枝。
class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
self.res = []
candidates.sort()
def dfs(tar, combination):
if tar == 0:
self.res.append(combination)
return
for candidate in candidates:
# 剪枝
if tar < candidate:
break
# 如果发现重复内容
if combination and candidate < combination[-1]:
continue
dfs(tar - candidate, combination + [candidate])
dfs(target, [])
return self.res
二解
不对原数组进行排序,与“一解”思路相同。
class Solution:
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
self.res = []
def dfs(begin, tar, combination):
if tar == 0:
self.res.append(combination)
return
for index in range(begin, n):
if tar < candidates[index]:
continue
dfs(index, tar - candidates[index], combination + [candidates[index]])
n = len(candidates)
dfs(0, target, [])
return self.res