回溯算法之组合问题Ⅱ

题目链接

给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用 一次 。

注意:解集不能包含重复的组合。 

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
输出:
[
[1,2,2],
[5]
]

本题是经典的组合问题,考虑使用回溯算法的模版加以修改即可,有两个要点

1、解集不能包含重复的组合。 

2、candidates 中的每个数字在每个组合中只能使用 一次 。

在遍历时要对这两种情况进行处理。

进入回溯三部曲(递归函数参数,递归终止条件,单层搜索逻辑)

1、递归函数参数

vector<vector<int>> res;
vector<int> path;
void backtracking(vector<int>& candidates, int target, int sum, int start)

其中res是我们返回的结果集,path是我们用来存储搜索到的所有可能的结果、sum是每次累加之后的结果,start表示从哪里开始取元素。

2、递归终止条件

显然当累加和sum等于target时终止循环,并将此时的结果集存入res

if (sum == target) {
    res.push_back(path);
    return;
}

3、单层搜索逻辑

首先进行一下剪枝操作,当sum+当前的candidates[i]大于target时(已排序),表示后面也不会再有合适的结果集了,可以直接跳出。

再进行去重操作,即相等的元素只使用一次,只需判断前后元素是否相等,相等则跳出此次循环

下面就是回溯的过程了:

sum += candidates[i];
path.push_back(candidates[i]);
backtracking(candidates, target, sum, i + 1);
sum -= candidates[i];
path.pop_back();

假如数据为

[10,1,2,7,6,1,5]和 8

排序后的数组为:[1,1,2,5,6,7,10]

假设此时path内元素为1,1;此时将2push后将5push发现比target大,返回到push5的循环中,此后6,7,10都将跳出,然后回到push2的循环中,将2给pop,push5发现和目标值相等,添加并返回,此后6,7,10都将不满足跳出。然后回到第二个1的循环中,然后将2,5,6,7,10都和第一个1组合一次,回溯完成。

for (int i = start; i < candidates.size(); i++) {
            if (sum + candidates[i] > target) {
                return;
            }
            if (i > start && candidates[i] == candidates[i-1]) {
                continue; // 跳过重复的候选数
            }
            sum += candidates[i];
            path.push_back(candidates[i]);
            backtracking(candidates, target, sum, i + 1);
            sum -= candidates[i];
            path.pop_back();
}

整体C++代码如下

class Solution {

public:

    vector<vector<int>> res;

    vector<int> path;

    void backtracking(vector<int>& candidates, int target, int sum, int start) {

        if (sum == target) {

            res.push_back(path);

            return;

        }

        for (int i = start; i < candidates.size(); i++) {

            if (sum + candidates[i] > target) {

                return;

            }

            if (i > start && candidates[i] == candidates[i-1]) {

                continue; // 跳过重复的候选数

            }

            sum += candidates[i];

            path.push_back(candidates[i]);

            backtracking(candidates, target, sum, i + 1);

            sum -= candidates[i];

            path.pop_back();

        }

    }

    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {

        sort(candidates.begin(), candidates.end()); // 对候选数进行排序

        backtracking(candidates, target, 0, 0);

        return res;

    }

};

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值