dfs回溯+剪枝+去除相同解+无相同数字每个数字用多次/有相同数字每个数字用一次 39. 组合总和 40. 组合总和 II

最新推荐文章于 2024-04-02 22:47:13 发布

豌豆射手GCC

最新推荐文章于 2024-04-02 22:47:13 发布

阅读量172

点赞数

分类专栏： leetcode DFS

本文链接：https://blog.csdn.net/BLUEsang/article/details/105960760

版权

leetcode 同时被 2 个专栏收录

189 篇文章 0 订阅

订阅专栏

DFS

24 篇文章 1 订阅

订阅专栏

本文介绍了如何使用深度优先搜索（DFS）结合回溯法解决组合总和的问题，针对39题和40题，分别给出了解题思路。在39题中，通过记录当前索引来避免重复组合；在40题中，首先对数组进行排序，然后利用当前索引避免重复，并在遇到重复数字时仅保留第一个。同时，文章还提到了剪枝策略以提高算法效率。

摘要由CSDN通过智能技术生成

39. 组合总和

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

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

说明：

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

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

示例 2:

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

解题
去重方法：记录当前索引，每次遍历下一组时从当前索引开始，即可去掉重复答案；

class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        dfs(candidates,0,0,target);
        return res;
    }

private:
    vector<vector<int>> res;
    vector<int> tmp;
    void dfs(vector<int> &candidates,int num,int j,int target)
    {
        if(num>target) return;
        if(num==target) res.push_back(tmp);
        for(int i=j;i<candidates.size();i++)
        {
            tmp.push_back(candidates[i]);
            dfs(candidates,num+candidates[i],i,target);
            tmp.pop_back();
        }
        return ;
    }
};

40. 组合总和 II

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

解题
先将数组排序，每次遍历记录当前索引，每次从当前的下一个数开始遍历，且若有重复的数时，只取第一个数，跳过后面的数；

剪枝条件： if(i>j&&candidates[i]==candidates[i-1]) continue;

class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end());
        visited.resize(candidates.size(),0);
        dfs(candidates,target,0,0);
        return res;
    }
private:
    vector<vector<int>> res;
    vector<int> tmp;
    void dfs(vector<int>& candidates,int target,int num,int j)
    {
        if(j>candidates.size()) return;
        if(num>target) return ;
        if(num==target) res.push_back(tmp);
        for(int i=j;i<candidates.size();i++)
        {
            if(i>j&&candidates[i]==candidates[i-1]) continue;  //减去的元素为非前第一个元素的相同元素
                tmp.push_back(candidates[i]);
                dfs(candidates,target,num+candidates[i],i+1);
                tmp.pop_back();
        }
        return;
    }
};