代码随想录算法训练营Day26

回溯算法理论基础

• 组合问题：N个数里面按一定规则找出k个数的集合
• 切割问题：一个字符串按一定规则有几种切割方式
• 子集问题：一个N个数的集合里有多少符合条件的子集
• 排列问题：N个数按一定规则全排列，有几种排列方式
• 棋盘问题：N皇后，解数独等等

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

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

第77题. 组合

力扣题目链接

n相当于宽度，k相当于高度

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> path = new ArrayList<>();
    public List<List<Integer>> combine(int n, int k) {
        backtracking(n , k , 1);
        return res;
    }
    public 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.remove(path.size()-1);
        }
    }
}

---

剪枝

1. 已经选择的元素个数：path.size();

2. 所需需要的元素个数为: k - path.size();

3. 列表中剩余元素（n-i） >= 所需需要的元素个数（k - path.size()）

4. 在集合n中至多要从该起始位置 : i <= n - (k - path.size()) + 1，开始遍历

为什么有个+1呢，因为包括起始位置，我们要是一个左闭的集合。

 

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> path = new ArrayList<>();
    public List<List<Integer>> combine(int n, int k) {
        backtracking(n , k , 1);
        return res;
    }
    public 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.remove(path.size()-1);
        }
    }
}