组合

题目

给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。

思路

简单的回溯算法的使用

代码

    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> res = new ArrayList<>();
        if (k <= 0 || n < k) {
            return res;
        }
        Deque<Integer> path = new ArrayDeque<>();
        dfs(n, k, 1, path, res);
        return res;
    }

    private void dfs(int n, int k, int begin, Deque<Integer> path, List<List<Integer>> res) {
        if (path.size() == k) {
            res.add(new ArrayList<>(path));
            return;
        }
        for (int i = begin; i <= n - (k - path.size()) + 1; i++) { // i <= n - (k - path.size()) + 1，剩余数据数量不足直接剪枝
            path.addLast(i);
            dfs(n, k, i + 1, path, res);
            path.removeLast();
        }
    }