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

回溯——组合

思路：

套了K次的循环，如何对K次循环呢，就需要一个startIndex，用它来控制从哪里开始

代码：

void combine_backTrack(int n, int k, int startIndex, vector<int>& path, vector<vector<int>>& result)
{
    //中止条件
    if (path.size() == k)
    {
        result.push_back(path);
        return;
    }

    for (int i = startIndex; i <= n; i++)
    {
        path.push_back(i);
        combine_backTrack(n, k, i + 1, path, result);
        path.pop_back();
    }
}

vector<vector<int>> combine(int n, int k) {
    vector<vector<int>> result;
    if (n == 0) return result;
    vector<int> path;    
    combine_backTrack(n, k, 1, path, result);
    return result;
}

 剪枝操作：

需要n-(k-p.size())+1

代码：

void combine_backTrack(int n, int k, int startIndex, vector<int>& path, vector<vector<int>>& result)
{
    //中止条件
    if (path.size() == k)
    {
        result.push_back(path);
        return;
    }

    for (int i = startIndex; i <= (n-(k-path.size())+1); i++)
    {
        path.push_back(i);
        combine_backTrack(n, k, i + 1, path, result);
        path.pop_back();
    }
}

vector<vector<int>> combine(int n, int k) {
    vector<vector<int>> result;
    if (n == 0) return result;
    vector<int> path;    
    combine_backTrack(n, k, 1, path, result);
    return result;
}