216. 组合总和 III/C++

216. 组合总和 III/C++_第1张图片

class Solution {
    vector<vector<int>> res;
    vector<int> vec;
    
    //num表示还可以选多少个数
    //target表示距离n还差多少
    //index表示从第几开始处理(由于不允许重复)
    void combination(int num,int target,int index){
    	//只有当数量与目标都为0时,才记录
        if(num==0 && target==0){
            res.push_back(vec);
            return;
        }
        //数量不足或者目标超过,都失败
        else if(num<=0 || target<=0)
            return;
        else{
        	//从index开始处理直到9为止
            for(int i=index;i<=9;++i){
                vec.push_back(i);
                //可选数字减一,目标减i,下次要从i+1开始处理
                combination(num-1,target-i,i+1);
                vec.pop_back();
            }
        }
    }
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        res.clear();
        if(k<=0 || n<=0)
            return res;
        
        combination(k,n,1);
        return res;
    }
};

你可能感兴趣的:(回溯法,LeetCode/C++)