代码随想录算法训练营第二十四天|77. 组合

77. 组合  

题目链接：力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台
文档讲解：代码随想录
视频讲解：带你学透回溯算法-组合问题（对应力扣题目：77.组合）| 回溯法精讲！_哔哩哔哩_bilibili

思路：本题是回溯法的经典题目。直接的解法当然是使用for循环，例如示例中k为2，很容易想到 用两个for循环，这样就可以输出 和示例中一样的结果。同时，本题可以使用剪枝优化，可以剪枝的地方就在递归中每一层的for循环所选择的起始位置。如果for循环选择的起始位置之后的元素个数 已经不足 我们需要的元素个数了，那么就没有必要搜索了。

C++代码：

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backTrack(int n, int k, int start){
        if(path.size() == k){
            result.push_back(path);
            return;
        }
        for(int i = start; i <= n; i++){
            path.push_back(i);
            backTrack(n, k, i + 1);
            path.pop_back();
        }
    }
    vector<vector<int>> combine(int n, int k) {
        backTrack(n, k, 1);
        return result;
    }
};

//剪枝
class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backTrack(int n, int k, int start){
        if(path.size() == k){
            result.push_back(path);
            return;
        }
        for(int i = start; i <= n - (k - path.size()) + 1; i++){
            path.push_back(i);
            backTrack(n, k, i + 1);
            path.pop_back();
        }
    }
    vector<vector<int>> combine(int n, int k) {
        backTrack(n, k, 1);
        return result;
    }
};