找出所有相加之和为 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