【玩转回溯算法专题】77——组合【中等】

1.题目链接

https://leetcode.cn/problems/combinations/description/

2.题目描述

给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。

示例 1：
ini 代码解读复制代码输入： n = 4, k = 2
输出：
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]

示例 2：
lua 代码解读复制代码输入： n = 1, k = 1
输出： [[1]]

提示：

1 <= n <= 20
1 <= k <= n

3.题目分析

递归函数的返回值以及参数

在这里要定义两个全局变量，一个用来存放符合条件单一结果，一个用来存放符合条件结果的集合。
代码如下：
ini 代码解读复制代码List<List> res = new ArrayList();// 存放符合条件结果的集合

List path = new ArrayList();// 用来存放符合条件结果

函数里一定有两个参数，既然是集合n里面取k个数，那么n和k是两个int型的参数。
然后还需要一个参数，为int型变量startIndex，这个参数用来记录本层递归的中，集合从哪里开始遍历（集合就是[1,…,n] ）。

4.代码实现

Java版本

    
    //定义全局返回值
    List<List<Integer>> res = new ArrayList();
    List<Integer> path = new ArrayList();

    public List<List<Integer>> combine(int n, int k) {
        //调用方法
        backing(n,k,1);
        return res;
        
    }

    public void backing(int n,int k,int startIndex){
        //终止条件
        if(path.size() == k){
            //符合条件存入res
            res.add(new ArrayList(path));
            return; 
        }

        //for 循环k  长度为n
        for(int i = startIndex;i<=n;i++){
            //添加
            path.add(i);
            //递归调用
            backing(n,k,i+1);
            //回溯
            path.removeLast();
        }
    
    }
}

剪枝操作

回溯法虽然是暴力搜索，但也有时候可以有点剪枝优化一下的。
优化过程如下：

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

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

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

在集合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) {
        //调用方法
        backing(n,k,1);
        return res;
        
    }

    public void backing(int n,int k,int startIndex){
        //终止条件
        if(path.size() == k){
            //符合条件存入res
            res.add(new ArrayList(path));
            return; 
        }

        //for 循环k  长度为n
        for(int i = startIndex;i<=n-(k-path.size()) + 1;i++){
            //添加
            path.add(i);
            //递归调用
            backing(n,k,i+1);
            //回溯
            path.removeLast();
        }
    
    }
}

C++版本

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(int n, int k, int startIndex) {
        if (path.size() == k) {
            result.push_back(path);
            return;
        }
        for (int i = startIndex; i <= n - (k - path.size()) + 1; i++) { // 优化的地方
            path.push_back(i); // 处理节点
            backtracking(n, k, i + 1);
            path.pop_back(); // 回溯，撤销处理的节点
        }
    }
public:

    vector<vector<int>> combine(int n, int k) {
        backtracking(n, k, 1);
        return result;
    }
};