题目:
找出所有相加之和为 n 的 k 个数的组合。
组合中只允许含有 1 - 9 的正整数,
并且每种组合中不存在重复的数字。
说明:
所有数字都是正整数。
解集不能包含重复的组合。
示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
思路:
回溯
解答:
/**
* @param {number} k
* @param {number} n
* @return {number[][]}
*/
var combinationSum3 = function(k, target) {
let s = '123456789';
let path = [];
let res = [];
let startIndex = 0;
backtrack(s, startIndex, k, target, path, res);
return res;
};
function backtrack(s, startIndex, k, target, path, res){
// 结束条件:
// 注意,这儿要是>,因为最后一个数可以加入path,下一层进来时刚好=s.length
if(startIndex > s.length || path.length > k){
return;
}
// 如果path个数为k,并且和为target
let sum = 0;
path.forEach(p => {
sum += p;
})
if(path.length === k && sum === target){
let pathCopy = [...path];
res.push(pathCopy);
return;
}
// 遍历所有选择
for(let i = startIndex;i < s.length; i++){
let num = s.charAt(i) - 0;
if(path.includes(num)){
continue;
}
// 不包含才做选择
path.push(num);
backtrack(s, i + 1, k, target, path, res);
path.pop();
}
}