初学python记录:力扣39. 组合总和

文章描述了一个编程问题,如何找出整数数组candidates中所有和为目标数target的不同组合,且避免重复计数。通过构建树结构并采用优化策略,确保只返回顺序不同的组合。作者提供了两种解决方案,一种是简单的递归方法,另一种是改进后的版本以去除重复组合。
摘要由CSDN通过智能技术生成

题目:

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。 

对于给定的输入,保证和为 target 的不同组合数少于 150 个。

提示:

  • 1 <= candidates.length <= 30
  • 2 <= candidates[i] <= 40
  • candidates 的所有元素 互不相同
  • 1 <= target <= 40

思考:

因为要找到所有和为target的组合,可以构建一棵树,根节点为target,每条路径的值为candidates中的数,路径连接的子节点即为 父节点-路径值 ,保证只有所有叶子节点的值为0(从根节点到叶子节点0的路径即为所求的一个组合)小于0(这条路径代表的组合不能满足和为target的条件)。以示例1为例,上述思路如下图所示:

可以看到找出了所有满足条件的组合,代码如下:

class Solution(object):
    def combinationSum(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        n = len(candidates)
        ans = []
        candidates.sort()

        def makeTree(target, candidates, path, ans):
            if target < 0:
                return
            elif target == 0:
                ans.append(path)
                return
            else:
                for index in range(0, n):
                    makeTree(target - candidates[index], candidates, path + [candidates[index]], ans)

        makeTree(target, candidates, [], ans)
        return ans

但是可以看到结果中出现了顺序不同,但元素完全相同的组合,但题目中说明这种组合是同样的,不应该重复返回。在图中用绿色表示应该删去的分支:

 

可以总结出:每一层的第二个节点生成子节点的路径值不能使用生成他的前序兄弟节点的路径值。

这样就避免了重复组合的出现,代码如下:

class Solution(object):
    def combinationSum(self, candidates, target):
        """
        :type candidates: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        n = len(candidates)
        ans = []
        candidates.sort()

        def makeTree(target, candidates, path, ans, begin_index):
            if target < 0:
                return
            elif target == 0:
                ans.append(path)
                return
            else:
                for index in range(begin_index, n):
                    makeTree(target - candidates[index], candidates, path + [candidates[index]], ans, index)

        makeTree(target, candidates, [], ans, 0)
        return ans

提交通过:

  • 7
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值