Leetcode216. 组合总和 III

题目

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。

分析

给定数字集合，找满足条件的组合。依旧是回溯问题，按照板子写就可，脑中一定要有回溯的搜索树的图形

代码

class Solution {
public:
    vector<int>path;
    vector<vector<int>>res;
    void backtracking(int n,int k,int startIndex,int sum){
        if(path.size() == k ){
            if(sum == n){
                res.push_back(path);
            } 
            return;
        }
        for(int i = startIndex; i <= 9;i++){
            path.push_back(i);
            sum += i;
            backtracking(n,k,i+1,sum);
            sum -= path.back();  //或者sum -= i;
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(n,k,1,0);
        return res;
    }
};

优化搜索，进行剪枝

 就是直接剪掉已选元素之和大于目标target的分支

class Solution {
public:
    vector<int>path;
    vector<vector<int>>res;
    void backtracking(int n,int k,int startIndex,int sum){
        if(sum > n){  //剪枝优化
            return;
        }
        if(path.size() == k ){
            if(sum == n){
                res.push_back(path);
            } 
            return;
        }
        for(int i = startIndex; i <= 9;i++){
            path.push_back(i);
            sum += i;
            backtracking(n,k,i+1,sum);
            sum -= path.back();  //或者sum -= i;
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(n,k,1,0);
        return res;
    }
};