原题链接:https://leetcode-cn.com/problems/combinations/
解题思路:
- 该题解参考了78. 子集的题解LeetCode题解:78. 子集,递归回溯,JavaScript,详细注释。
- 该题可以理解为,在1…n中,取长度为k的子集。
- current代表了当前值,它有取或者不取两种状态。将其存入subset,并进入下层递归,代表取值状态。将其从subset中弹出,再进入下层递归,表示不取值的状态。
- 递归终止条件有两个,即subset达到所需长度k,或者当剩余可选的元素数量少于subset还需要的元素数量。
function dfs(result, subset, current, n, k) {
// 递归终止条件
// 当生成的子集长度等于组合长度,则存储结果并退出递归
if (subset.length === k) {
result.push(subset.slice());
return;
}
// 进行剪枝,当剩余可选的元素数量少于subset还需要的元素数量时,退出递归
// 由于当前元素是可选的,因此需要加1。
if (k - subset.length > n - current + 1) {
return;
}
// 将当前值存入子集,并继续递归
subset.push(current);
dfs(result, subset, current + 1, n, k);
// 将当前值取出,以当前值未被存储的状态,进入下一个数的递归
subset.pop();
dfs(result, subset, current + 1, n, k);
}
/**
* @param {number} n
* @param {number} k
* @return {number[][]}
*/
var combine = function (n, k) {
let result = []; // 存储结果
let subset = []; // 存储每个子集
// 使用DFS+回溯生成所有子集
// 从1开始存储子集,因此current初始值为1
dfs(result, subset, 1, n, k);
return result;
};