【leetcode】#数组【Python】39. Combination Sum 组合总和回溯

最新推荐文章于 2022-11-23 12:18:42 发布

风轻扬逍遥子

最新推荐文章于 2022-11-23 12:18:42 发布

阅读量439

点赞数

CC 4.0 BY-SA版权

分类专栏：【leetcode】#数组【leetcode】【#数组】【Python】文章标签： leetcode python 39 Combination Sum 组合总和

本文链接：https://blog.youkuaiyun.com/u014256231/article/details/83721018

【leetcode】#数组同时被 2 个专栏收录

42 篇文章

订阅专栏

【leetcode】【#数组】【Python】

37 篇文章

订阅专栏

这篇博客介绍了LeetCode第39题——组合总和的解题思路，特别是如何使用Python实现回溯算法，并通过优化减少了函数调用次数，提升了效率。博主分享了标准解法以及自己改进后的高效解法，该解法在44ms内完成了运算，超越了99.8%的提交。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

链接：

https://leetcode-cn.com/problems/combination-sum/

题目：

给定一个无重复元素的数组 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(object):
    def combinationSum(self, candidates, target):
        res = []
        candidates.sort()
        self.dfs(candidates, target, 0, [], res)
        return res
    
    def dfs(self, nums, target, index, path, res):
        if target < 0:
            return  # backtracking
        if target == 0:
            res.append(path)
            return 
        for i in xrange(index, len(nums)):
            self.dfs(nums, target-nums[i], i, path+[nums[i]], res)

我的改进解法：44ms，超越99.8%。把判断放在循环里，更少调用函数

class Solution(object):
    def combinationSum(self, candidates, target):
        candidates.sort()                      
        result = []
        self.combine_sum(candidates, 0, [], result, target)
        return result

    def combine_sum(self, nums, start, path, result, target):
        if target == 0:
            result.append(path)
            return
        for i in range(start, len(nums)):
            if nums[i] > target:
                break
                return
            self.combine_sum(nums, i, path+[nums[i]], result, target-nums[i])