给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例 :
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
/**
* @param {number} n
* @param {number} k
* @return {number[][]}
*/
// 回溯
var combine = function(n, k) {
let res = []
let path = []
const backTracking = function (n, k, startIndex){
if(path.length === k){
res.push([...path])
return
}
for(let i = startIndex; i <= n; i++){
path.push(i)
backTracking(n, k, i+1)
path.pop()
}
}
backTracking(n, k, 1);
return res
};
剪枝
当 i
取一个值时,path
中已经有了 path.length
个元素,还需 k-path.length
个。可取的数字还剩下 n-i+1
个所以要 n-i+1 >= k-path.length
所以循环里改成 i<= n-(k-path.length)+1
就行