组合总数问题——LeetCode 39 40

题意:

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

candidates 中的数字可以无限制重复被选取。

说明:

所有数字(包括 target)都是正整数。
解集不能包含重复的组合。 
示例 1:

输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
  [7],
  [2,2,3]
]

这一题是非常典型的回溯法的题目:

回溯法的核心思想就是在当前状态下不断找到下一个满足递归条件的状态,然后递归,若满足递归终止条件,那么就将递归的路径输出,看到这里会发现,和树的深度优先遍历非常相似,实际上这就是一个树的深度优先遍历,不过在遍历的过程中要保存遍历的路径,同时要保证高效的话,还要对决策进行“剪枝”,以避免不必要的递归。

本题的代码如下:

class Solution {
private:
    vector<vector<int>> res;
    vector<int> path;
    vector<int> candidates;
public:
    
    void helper(int start,int target){
        if (target<0){
            return;
        }else if(0==target){
            res.push_back(path);
        }else{
            for(int i=start;i<candidates.size();i++){
                if(candidates[i]<=target){
                    path.push_back(candidates[i]);
                    helper(i,target-candidates[i]);
                    path.pop_back();
                }
            }
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        
        vector<int> path;
        this->candidates=candidates;
        helper(0,target);
        return res;
    }
};

另外注意到,此题中的candidates数组中的元素都可以无限次使用,若该数组中的所有元素都只能使用一次的话,那么就转换成了如下问题:

给定一个数组 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]
]

这里candidate数组中的每个元素都只能使用一次,因此,在递归的时候就不能够像上题中直接使用本次递归所使用的索引i,而要使用下一个索引,以下是本题代码:

#include <iostream>
#include <vector>

using namespace std;

class Solution {

private:
    vector<int> candidates;
    vector<vector<int>> res;
    vector<int> path;
public:
    void DFS(int start, int target) {
        if (target == 0) {
            res.push_back(path);
            return;
        }

        for (int i = start; i < candidates.size() && target - candidates[i] >= 0; i++) {
            if (i > start && candidates[i] == candidates[i - 1])
                continue;
            path.push_back(candidates[i]);
            // 元素不可重复利用,使用下一个即i+1
            DFS(i + 1, target - candidates[i]);
            path.pop_back();
        }
    }

    vector<vector<int>> combinationSum2(vector<int> &candidates, int target) {
        sort(candidates.begin(), candidates.end());
        //这里对candidate数组进行了排序,可以有效地避免重复的情况出现,
        //也就是避免了将路径加入res中时要先检查res中是否已经存在了该路径。
        this->candidates = candidates;
        DFS(0, target);
        return res;
    }
};

上述代码使用了一个非常巧妙的技巧:先对candidate数组进行排序,然后再对齐进行递归。这样可以有效避免重复路径的情况出现,也就是避免了将路径加入res中时要先检查res中是否已经存在了该路径。


参考文章:https://leetcode-cn.com/problems/combination-sum-ii/solution/hui-su-suan-fa-jian-zhi-python-dai-ma-java-dai-m-3/

 

 

 

 

 

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值