代码随想录算法训练营第二十四天 | 力扣 77. 组合

回溯

回溯算法是一种经典的搜索算法，通常用于在有限状态空间中找到所有或满足特定条件的解决方案。

回溯算法通过尝试所有可能的解决方案，直到找到一个满足条件的解决方案或者所有的可能解决方案都被尝试过后，回溯到之前的状态，继续尝试其他的解决方案。

77. 组合

题目

77. 组合

给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。

你可以按 任何顺序 返回答案。

解析

使用回溯算法来解，根据题意终止条件是所需的组合数是k，将此组合放到最终的结果集中，遍历条件就是从，一个startIndex开始，startIndex的作用是去重，在前面取过的数后面就不能再取了，然后还在循环条件做了剪枝操作 n - (k - path.size()) + 1 ，就是在剩余的数不足以再构造成为所需的结果，就不需要再去遍历了。k是所需的数量，path.size()是已经存在的数量，(k - path.size())是还需要的数量，所以遍历的开始的下标最多到 n - (k - path.size()) + 1，+1是要包含左边界。

Java代码实现

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

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

private void combination(int n, int k, int startIndex) {
    if (path.size() == k) {
        result.add(new ArrayList<>(path));
        return;
    }
    for (int i = startIndex; i <= n - (k - path.size()) + 1; i++) {
        path.add(i);
        combination(n, k, i + 1);
        // 回溯
        path.removeLast();
    }
}