零基础代码随想录【Day24】|| 回溯算法理论基础，77. 组合

目录

DAY24

回溯算法理论基础

第77题. 组合

解题思路&代码

DAY24

回溯算法理论基础

回溯三部曲。

• 回溯函数模板返回值以及参数

在回溯算法中，我的习惯是函数起名字为backtracking，这个起名大家随意。

回溯算法中函数返回值一般为void。

再来看一下参数，因为回溯算法需要的参数可不像二叉树递归的时候那么容易一次性确定下来，所以一般是先写逻辑，然后需要什么参数，就填什么参数。

• 回溯函数终止条件

既然是树形结构，那么我们在讲解二叉树的递归 (opens new window)的时候，就知道遍历树形结构一定要有终止条件。

所以回溯也有要终止条件。

什么时候达到了终止条件，树中就可以看出，一般来说搜到叶子节点了，也就找到了满足条件的一条答案，把这个答案存放起来，并结束本层递归。

• 回溯搜索的遍历过程

在上面我们提到了，回溯法一般是在集合中递归搜索，集合的大小构成了树的宽度，递归的深度构成的树的深度。

第77题. 组合

力扣题目链接(opens new window)

给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。

示例: 输入: n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]

解题思路&代码

思路：

可以看出这棵树，一开始集合是 1，2，3，4， 从左向右取数，取过的数，不再重复取。

第一次取1，集合变为2，3，4 ，因为k为2，我们只需要再取一个数就可以了，分别取2，3，4，得到集合[1,2] [1,3] [1,4]，以此类推。

每次从集合中选取元素，可选择的范围随着选择的进行而收缩，调整可选择的范围。

图中可以发现n相当于树的宽度，k相当于树的深度。

图中每次搜索到了叶子节点，我们就找到了一个结果。

相当于只需要把达到叶子节点的结果收集起来，就可以求得 n个数中k个数的组合集合

class Solution {
    List<List<Integer>> result= new ArrayList<>();//二维数组储存最终的多个组合结果
    LinkedList<Integer> path = new LinkedList<>();
    public List<List<Integer>> combine(int n, int k) {
        backtracking(n,k,1);
        return result;
    }

    public void backtracking(int n,int k,int startIndex){
        if (path.size() == k){//path是存放k组合结果的地方，当path的大小为k时，说明找到了满足条件的组合，这里是终止条件
            result.add(new ArrayList<>(path));
            return;
        }
        for (int i =startIndex;i<=n;i++){//startIndex不断更新，这里是单层逻辑，分别针对[1, 2],[2, 3]的情况，每个不同的i代表不同的起点
            path.add(i);
            backtracking(n,k,i+1);
            path.removeLast();/** 回溯，回撤一下之前的1，2，开始找1，3，例如i=2的情况下，满足条件，
            所以return了，也就跳出了backtracking(n,k,2+1），再所以删除了2，然后开始i为3的遍历，但是最开始的遍历是i=1*/
        }
    }
}