Bootstrap

【LC】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

题解:

class Solution {
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> res = new ArrayList<>();  // 存储最终结果的列表
        dfs(res, new ArrayList<>(), 1, n, k);  // 开始回溯
        return res;
    }

    private void dfs(List<List<Integer>> res, List<Integer> path, int start, int n, int k) {
        int d = k - path.size();//还需选d个数
        if (d == 0) {  // 已经选好d个数,找到一种组合
            res.add(new ArrayList<>(path));  // 将当前路径的副本添加到结果列表中
            return;
        }
        for (int i = start; i <= n && n - i + 1 >= d; i++) {
            path.add(i);  // 将元素添加到路径中
            dfs(res, path, i + 1, n, k);  // 递归调用,更新起始位置
            path.remove(path.size() - 1);  // 回溯操作,将元素从路径中移除
        }
    }
}

;