77. 组合
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
示例 2:
输入:n = 1, k = 1
输出:[[1]]
提示:
1 <= n <= 20
1 <= k <= n
思路:
使用回溯法:
使用path存放叶子结点,res收集所有叶子节点
回溯函数的参数为起始遍历索引startIndex和记录遍历个数index,
当index等于k,遍历结束收集叶子节点;由于是组合问题,所以每次遍历要从下一个开始,startIndex为i+1;path存放中间结果,递归,最后回溯。
代码:
/**
* @param {number} n
* @param {number} k
* @return {number[][]}
*/
var combine = function (n, k) {
const res = [];
const path = [];
backtracking(0, 0);
return res;
function backtracking(startIndex, index) {
if (index === k) {
res.push([...path]);
return;
}
for (let i = startIndex; i < n; i++) {
path.push(i + 1);
backtracking(i + 1, index + 1);
path.pop();
}
}
};