代码随想录-71-77. 组合

前言

我在刷卡哥的“代码随想录”，自己的总结笔记均会放在“算法刷题-代码随想录”该专栏下。
代码随想录此题链接

题目

1.组合问题（回溯+剪枝）

经典的组合问题，通过回溯模板写此题

全局变量

//1.最终总的结果集
ArrayList<ArrayList<Integer>> result = 
new ArrayList<ArrayList<Integer>>();//返回的结果
//2.每次组合的集合
ArrayList<Integer> path = new ArrayList<Integer>();

2. 本题思路分析：

回溯三部曲：

1. 参数与返回值：

• 参数：n，k，startIndex（n是for循环的范围（是n叉树的宽度），k是每一个组合集标准大小(是n叉树的深度)，startIndex是下一次循环的开始节点）
• 返回值：void

2. 终止条件：当组合集大小等于标准组合集大小，说明此时当前的组合集已经满足条件了，结束此次递归。
3. 单层循环逻辑：

• 从当前的startIndex开始，到n结束循环遍历；
• 开始将当前值加入组合集合
• 将i+1设置成startIndex，开始递归。
• 递归完成后回溯，把当前值从组合集合中去除。

3. 算法实现

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

    public void backtracking(int n,int k,int statrtIndex){
        if(path.size() == k){
            ArrayList<Integer> tmp = new ArrayList<>(path);//通过一个ArrayList创建另一个包含元素完全相同的ArrayList对象
            result.add(tmp);
            return;
        }
        for(int i = statrtIndex;i <= n - (k - path.size()) + 1;i++){//剪枝代码部分
            path.add(i);
            backtracking(n,k,i + 1);
            path.remove(path.size() - 1);
        }
    }
}

4. 算法坑点

1. 可以剪枝优化算法效率
   原始未优化前

 for(int i = statrtIndex;i <= n;i++)

优化后。

 for(int i = statrtIndex;i <= n - (k - path.size()) + 1;i++)