搜索回溯—组合总和 II(leetcode 40)

题目描述

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

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

说明:

    所有数字(包括目标数)都是正整数。
    解集不能包含重复的组合。

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
  [1, 7],
  [1, 2, 5],
  [2, 6],
  [1, 1, 6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
  [1,2,2],
  [5]
]

算法分析

方法一:回溯

思路与算法

由于我们需要求出所有和为 target的组合,并且每个数只能使用一次,因此我们可以使用递归 + 回溯的方法来解决这个问题:

    我们用 dfs(pos,rest), 表示递归的函数,其中 pos表示我们当前递归到了数组 candidates中的第 pos个数,而 rest表示我们还需要选择和为 rest的数放入列表作为一个组合;

    对于当前的第 pos个数,我们有两种方法:选或者不选。如果我们选了这个数,那么我们调用 dfs(pos+1,rest−candidates[pos]), 进行递归,注意这里必须满足 rest≥candidates[pos]。如果我们不选这个数,那么我们调用 dfs(pos+1,rest)进行递归;

    在某次递归开始前,如果 rest的值为 0,说明我们找到了一个和为 target的组合,将其放入答案中。每次调用递归函数前,如果我们选了那个数,就需要将其放入列表的末尾,该列表中存储了我们选的所有数。在回溯时,如果我们选了那个数,就要将其从列表的末尾删除。

上述算法就是一个标准的递归 + 回溯算法,但是它并不适用于本题。这是因为题目描述中规定了解集不能包含重复的组合,而上述的算法中并没有去除重复的组合。

    例如当 candidates=[2,2],target=2 时,上述算法会将列表 [2]放入答案两次。

if cur > begin and candidates[cur-1] == candidates[cur]  // 避免重复的

代码

class Solution {
public:
    void dfs(vector<int>& candidates, vector<vector<int>> &ans, vector<int> &tmp, int index, int target) {
        
        if(target == 0) {
            ans.push_back(tmp);
            return;
        }
        for(int i = index; i < candidates.size() && target - candidates[i] >= 0; ++i) {
            if(i > index && candidates[i] == candidates[i-1]){
                continue;
            }
            tmp.push_back(candidates[i]);
            dfs(candidates, ans, tmp, i+1, target-candidates[i]);
            tmp.pop_back();
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<vector<int>> ans;
        vector<int> tmp;
        sort(candidates.begin(), candidates.end());
        dfs(candidates, ans, tmp, 0, target);
        return ans;
    }
};

时间复杂度分析

时间复杂度:O(2^n*n)

空间复杂度:O(n)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值