LeetCode题解:77. 组合,递归回溯,JavaScript,详细注释

211 阅读1分钟

原题链接:leetcode-cn.com/problems/co…

解题思路:

  1. 该题解参考了78. 子集的题解LeetCode题解:78. 子集,递归回溯,JavaScript,详细注释
  2. 该题可以理解为,在1...n中,取长度为k的子集。
  3. current代表了当前值,它有取或者不取两种状态。将其存入subset,并进入下层递归,代表取值状态。将其从subset中弹出,再进入下层递归,表示不取值的状态。
  4. 递归终止条件有两个,即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;
};