代码随想录算法训练营第二四天 | 回溯 组合

LeetCode 77. 组合

回溯模板

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

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

组合

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

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

class Solution {
    List<List<Integer>> result = new ArrayList<>();
    List<Integer> list = new ArrayList<>();
    public List<List<Integer>> combine(int n, int k) {
        backtracking(n, k, 1);
        return result;
    }

    // n相当于树的宽度，k相当于树的深度
    /**
     * 每次从集合中选取元素，可选择的范围随着选择的进行而收缩，调整可选择的范围，就是要靠startIndex
     * @param startIndex 用来记录本层递归的中，集合从哪里开始遍历（集合就是[1,...,n] ）。
     */
    private void backtracking(int n, int k, int startIndex) {
        if (list.size() == k) {   // 终止条件
            result.add(new ArrayList<>(list));
            return;
        }

        for (int i = startIndex; i <= n - (k - list.size()) + 1; i++) { // 剪枝 
            list.add(i);
            backtracking(n, k, i+1); // i+1  不是startIndex + 1 ,下次在后一个开始选数 
            list.removeLast();  // list.removeLast()
        }
    }
}