Skip to the content.

leetcode [39] 组合总和


Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub


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

思路:

递归解决子问题

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates = sorted(candidates)
        return self.combinationSumCore(candidates, target)

    def combinationSumCore(self, candidates, target: int):
        if len(candidates) == 0 or target <= 0:
            return []
        if len(candidates) == 1:
            if candidates[0] == target:
                return [[target]]
                
        result = []
        for i, c in enumerate(candidates):
            if c == target:
                result.append([c])
                continue
            if c > target:
                break
            for can in self.combinationSum(candidates[i:], target - c):
                result.append([c] + can)
        return result