leetcode39. 组合总和

最新推荐文章于 2024-03-08 08:30:00 发布

ShawDa

最新推荐文章于 2024-03-08 08:30:00 发布

阅读量890

点赞数 1

分类专栏： LeetCode

本文链接：https://blog.csdn.net/sinat_36811967/article/details/87455890

版权

LeetCode 专栏收录该内容

240 篇文章 1 订阅

订阅专栏

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

很显然需要用回溯的思想来解题，记录一个临时list保存某个组合，记录一个num保证组合不会重复：

class Solution:
    def combinationSum(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        if not candidates:
            return []
        candidates.sort()  # 先排序
        res = []
        self.helper(target, candidates, res, [], 0)  # []记录一个临时list,0记录位置保证不重复
        return res

    def helper(self, target, candidates, res, one, num):
        if target == 0:  # target为0说明刚好组合
            res.append(one)
            return
        if target < candidates[0]:  # 不可能有组合
            return
        for i in range(num, len(candidates)):
            if candidates[i] > target:
                break
            tmp = [i for i in one]  # 注意这里要copy
            tmp.append(candidates[i])
            self.helper(target-candidates[i], candidates, res, tmp, i)