回溯算法之组合求和

给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合,candidates 中的数字可以无限制重复被选取,示例如下:

数组 candidates 中不包含0,如果有0要考虑无限循环的情况;数组中不包含重复元素,若包含重复元素要考虑去重的问题。

本题与组合问题的区别是不限定组合元素的数量,并且可以重复选取。

回溯三部曲:

(1)参数和返回值:返回值为void,参数根据后面代码确定。

(2)返回条件:返回条件需要根据组合的和来确定,组合的和大于目标值就返回。

(3)单层搜索逻辑:树的结构搜索,树的宽度为集合个数,深度由返回条件确定。

实现代码如下:

//组合求和
vector<int> vec;
vector<vector<int>> result;
int sum = 0;

void backtracking(vector<int>& candidates, int start, int& target) {
    //确定终止条件
    if(sum == target) {
        result.push_back(vec);
        return;
    }
    else if(sum > target) {
        return;
    }
    //单层搜索的逻辑
    for(int i = start; i<candidates.size(); i++) {
        vec.push_back(candidates[i]);
        sum += candidates[i];
        backtracking(candidates, i, target); //递归
        sum -= candidates[i]; //回溯
        vec.pop_back();
    }
}

vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
    vec.clear();
    result.clear();
    backtracking(candidates, 0, target);
    return result;
}

4.增加剪枝操作

增加剪枝,如果加第 i 个元素已经大于目标值之后,就不再遍历后面的元素,但前提是需要对数组进行排序,实现代码如下:

//增加剪枝 
vector<int> vec;
vector<vector<int>> result;
int sum = 0;

void backtracking(vector<int>& candidates, int start, int& target) {
    //确定终止条件
    if(sum == target) {
        result.push_back(vec);
        return;
    }
    else if(sum > target) {
        return;
    }
    //单层搜索的逻辑 增加剪枝
	//增加剪枝,如果加i已经大于目标值之后,就不再使用i+1
	//前提是需要给数组进行排序 
    for(int i = start; i<candidates.size() && sum + candidates[i] <= target; i++) {
        vec.push_back(candidates[i]);
        sum += candidates[i];
        sorted(vec.begin(), vec.end()); 
        backtracking(candidates, i, target); //递归
        sum -= candidates[i]; //回溯
        vec.pop_back();
    }
}

vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
    vec.clear();
    result.clear();
    backtracking(candidates, 0, target);
    return result;
}

在求和问题中,排序之后再进行剪枝操作是常用的方法。

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值