力扣(216)组合总和Ⅲ

本文旨在记录解题过程中的收获,使用C++

与普通组合问题不同的地方在于要考虑组合中元素的和是多少,所以要引入targetSum,在元素压入栈之前,更新sum的值,递归结束后,要记得更新sum的值(即删除sum的值),将栈里面的元素弹出。

我的代码:

class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backTracking(n,k,0,1);
        return result;
    }
    void backTracking(int targetSum,int k,int sum,int startIndex){
        if(path.size()==k){
            if(targetSum==sum){
                result.push_back(path);
                return;
            }   
        }
        for(int i=startIndex;i<=9;i++){
            sum+=i;
            path.push_back(i);
            backTracking(targetSum,k,sum,i+1);
            sum-=i;
            path.pop_back();
        }

    }
    vector<int> path;
    vector<vector<int>> result;

};

优化思路:采取剪枝的方法进行优化

​
class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backTracking(n,k,0,1);
        return result;
    }
    void backTracking(int targetSum,int k,int sum,int startIndex){
        if(startIndex>targetSum){
            return;
        }
        if(path.size()==k){
            if(targetSum==sum){
                result.push_back(path);
                return;
            }   
        }
        for(int i=startIndex;i<=9-(k-path.size())+1;i++){
            sum+=i;
            path.push_back(i);
            backTracking(targetSum,k,sum,i+1);
            sum-=i;
            path.pop_back();
        }

    }
    vector<int> path;
    vector<vector<int>> result;

};

​

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值