39. 组合总和

给定一个无重复元素的正整数数组 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