代码随想录算法训练营第24天

第77题. 组合

力扣原题链接

回溯法解决的问题都可以抽象为树形结构（N叉树），把组合问题抽象为如下树形结构：

图中可以发现n相当于树的宽度，k相当于树的深度。

相当于只需要把达到叶子节点的结果收集起来，就可以求得 n个数中k个数的组合集合。

回溯模板:

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (选择：本层集合中元素（树中节点孩子的数量就是集合的大小）) {
        处理节点;
        backtracking(路径，选择列表); // 递归
        回溯，撤销处理结果
    }
}

题解：

class Solution {

    List<List<Integer>> res = new ArrayList<>();
    LinkedList<Integer> path = new LinkedList<>();

    public List<List<Integer>> combine(int n, int k) {
        backtracking(n, k, 1);
        return res;
    }

    private void backtracking(int n, int k, int startIndex) {
        if (path.size() == k) {
            res.add(new ArrayList<>(path));
            return;
        }
        for (int i = startIndex; i <= n; i++) {
            path.add(i);
            backtracking(n, k, i + 1);
            path.removeLast();
        }
    }
}

优化：做剪枝操作，这题是在回溯的 for 循环中的条件中做剪枝。

public static void main(String[] args) {
    Scanner in = new Scanner(System.in);
    combine_ com = new combine_();
    int n = in.nextInt();
    int k = in.nextInt();
    System.out.println(com.combine(n, k));
}

List<List<Integer>> res = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();

public List<List<Integer>> combine(int n, int k) {
    backtracking(n, k, 1);
    return res;
}

private void backtracking(int n, int k, int startIndex) {
    if (path.size() == k) {
        res.add(new ArrayList<>(path));
        return;
    }
    for (int i = startIndex; i <= n - (k - path.size()) + 1; i++) {  // 剪枝
        path.add(i);
        backtracking(n, k, i + 1);
        path.removeLast();
    }
}