刷代码随想录有感（67）：回溯算法——组合总和

最新推荐文章于 2024-07-13 23:05:26 发布

西南男子单身联盟

最新推荐文章于 2024-07-13 23:05:26 发布

阅读量188

点赞数 1

分类专栏：算法沉思文章标签：算法数据结构 leetcode

本文链接：https://blog.csdn.net/m0_74073296/article/details/138948700

版权

算法沉思专栏收录该内容

139 篇文章 2 订阅

订阅专栏

题干：

代码：

class Solution {
public:
    vector<int>tmp;
    vector<vector<int>>res;
    void backtracking(int k, int n, int sum, int start){
        if(tmp.size() == k && sum == n){
            res.push_back(tmp);
            return;
        }
        for(int i = start; i <= 9; i++){
            sum += i;
            tmp.push_back(i);
            backtracking(k, n, sum, i + 1);
            sum -= i;
            tmp.pop_back();
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(k, n, 0, 1);
        return res;
    }
};

剪枝：如果sum > n 则终止，如果取不到k个数则终止：

class Solution {
public:
    vector<int>tmp;
    vector<vector<int>>res;
    void backtracking(int k, int n, int sum, int start){
        if(sum > n)return;//剪枝1
        if(tmp.size() == k && sum == n){
            res.push_back(tmp);
            return;
        }
        for(int i = start; i <= 9 - (k - tmp.size()) + 1; i++){//剪枝2
            sum += i;
            tmp.push_back(i);
            backtracking(k, n, sum, i + 1);
            sum -= i;
            tmp.pop_back();
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(k, n, 0, 1);
        return res;
    }
};

回溯调用参数不是start而是i + 1...