【代码随想录训练营】Day24-回溯算法

代码随想录训练营 Day24

今日任务

回溯算法理论基础
77.组合

回溯算法理论基础

1. 回溯算法本质：是递归的副产品；本质是穷举，并非高效的算法；可以将回溯问题抽象为树结构
2. 可以解决的问题：
   ① 组合问题：没有顺序问题
   ② 切割问题：一个字符串按一定规则有几种切割方式
   ③ 子集问题：N个数的集合里有多少符合条件的子集
   ④ 排列问题：有顺序问题
   ⑤ 棋盘问题：N皇后，解数独
3. 模板
   ① 确定回溯函数参数及返回值
   ② 确定回溯函数终止条件
   ③ 确定回溯搜索的遍历过程

void backtracking(参数){
	if(终止条件){
		存放结果;
		return;
	}
	for(选择本层集合中元素){
		处理节点;
		backtracking(参数);
		回溯，撤销处理结果;
	}
}

77. 组合

链接：https://leetcode.cn/problems/combinations/
剪枝优化：还需要 k - path.size() 个元素，所以起始位置 i 最远可以到 n - (k - path.size()) + 1，形成一个闭区间，如果不加1会漏掉很多起始位置合法的情况。

class Solution {
    List<List<Integer>> result;
    List<Integer> path;
    public void findCombinition(int n, int k, int startIdx){
        if(path.size() == k){
            result.add(new ArrayList(path));
            return;
        }
        //for(int i = startIdx; i <= n; i++){
        for(int i = startIdx; i <= n - (k - path.size()) + 1; i++){ //剪枝优化
            path.add(i);
            findCombinition(n, k, i + 1); //画图，树形结构
            path.remove(path.size() - 1);
        }
    }
    public List<List<Integer>> combine(int n, int k) {
        result = new ArrayList<List<Integer>>();
        path = new ArrayList<Integer>();
        findCombinition(n, k, 1);
        return result;
    }
}