77.组合-CSDN博客

本文链接：https://blog.csdn.net/qq_46514141/article/details/125468956

77. 组合

在这里插入图片描述

这道题当然可以用for循环嵌套解决，但是嵌套层数要根据k变化，很难写，所以还是用回溯。

回溯的核心就是for循环横向展开，然后递归纵向展开，例如n=4， k=2，一开始的集合是[1,2,3,4]，然后for循环依次取1,2,3,4，取了之后就递归继续分解。

77.ç»„åˆ

图片来源：代码随想录。

回溯回来的时候一定要消除递归的影响，即把改变了的值改回来。

class Solution {
private:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(int n, int k, int position) {
        if (path.size() == k)
            result.push_back(path);
        for (int i = position; i <= n; 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;
        
    }
};

剪枝优化：

回溯一般可以剪枝，剪枝在递归的前面进行，判断要不要进入递归，这里的剪枝方法是判断剩下的数字够不够用了。k-path.size()是还需要的数字个数，n-i+1是仍有的数字个数，若前者大于后者直接break即可。

class Solution {
private:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(int n, int k, int position) {
        if (path.size() == k)
            result.push_back(path);
        for (int i = position; i <= n; i++) {
            if ((k - path.size() > (n - i + 1)))
                break;
            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;
        
    }
};