216. 组合总和 III

找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:

  • 只使用数字1到9
  • 每个数字 最多使用一次 

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
解释:
1 + 2 + 6 = 9
1 + 3 + 5 = 9
2 + 3 + 4 = 9
没有其他符合的组合了。

示例 3:

输入: k = 4, n = 1
输出: []
解释: 不存在有效的组合。
在[1,9]范围内使用4个不同的数字,我们可以得到的最小和是1+2+3+4 = 10,因为10 > 1,没有有效的组合。

提示:

  • 2 <= k <= 9
  • 1 <= n <= 60

思路:没有做过77.组合这道题的建议先做一下那道题,这道题是在那道题的基础上稍微做点改动即可。题解我放在下面了。那道题只需要在1~n中选出k个数即可,这道题是要在1~9中选出k个数,其和等于n,那就在77那道题的基础上,给回溯算法的终止条件多判断一个其和是否等于n,等于我才把他放入result中,这就完成了这道题。到此为止已经满足了题意,但是当数据量太大的时候会超时,所以我们得对上面的算法做剪枝,那么什么地方能改进呢,肯定是for循环的范围,可以想一下,我们是从1-9中取k个数,让他们的和等于n,若还没有取k个数,他的和已经大于n了,那根本不用进行后面的循环了,可以把后面的部分全部剪枝剪掉,这就省下了很多时间。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]

比如这个例子,已经取了数字1,2,第一轮递归backtacking,拿到的是3,1+2+3<7,进行第二轮递归backtacking,拿到的是4,1+2+4=7,其实已经不需要再拿后面的数了,后面的数5~9,任意一个加上1和2,和都大于7,所以这个地方可以进行剪枝,当然得保证这个数的范围在1~9,所以最终的范围是min(n-sum(path)+1,10),可能有点绕,但自己多捋捋,就会豁然开朗。

77.组合题解:77. 组合-CSDN博客

代码(Python):

class Solution(object):
    def combine(self, n, k):
        result = []                               #保存最终的结果
        path = []                                 #保存每个满足要求的结果
        def backtracking(n,k,startIndex):
            if len(path) == k:                    #如果path中有k个数了,就把这个结果放入result
                if sum(path) == n:
                    result.append(path[:])
                    return
            for x in range(startIndex,min(n-sum(path)+1,10)):
                path.append(x)                    
                backtracking(n,k,x+1)             #递归调用backtracking
                path.pop()                        #回溯
        backtracking(n,k,1)                       
        return result
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值