当前位置:   article > 正文

算法练习第22天|39. 组合总和、40.组合总和II

算法练习第22天|39. 组合总和、40.组合总和II

39. 组合总和

39. 组合总和 - 力扣(LeetCode)icon-default.png?t=N7T8https://leetcode.cn/problems/combination-sum/description/

题目描述:

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。 

对于给定的输入,保证和为 target 的不同组合数少于 150 个。

示例 1:

输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。

示例 2:

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

示例 3:

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

提示:

  • 1 <= candidates.length <= 30
  • 2 <= candidates[i] <= 40
  • candidates 的所有元素 互不相同
  • 1 <= target <= 40

 思路分析:

由于所有candidates的数>=1,所以不必担心有0的问题。另外,由于元素可以多次使用,所以其搜索过程对应的树形结构如下所示:

注意图中叶子节点的返回条件,因为本题没有组合数量要求,仅仅是总和的限制,所以递归没有层数的限制,只要选取的元素总和超过target,就返回! 另外,注意蓝色框中文字,因为要求的是满足条件的

下面我们结合上述的树形结构图,按照回溯三部曲来尝试书写代码:

  • 回溯第一步:确认回溯函数的参数和返回值。

这里依然是定义两个全局变量,二维数组result存放结果集,数组path存放符合条件的结果。(这两个变量可以作为函数参数传入) 

老规矩,返回值类型void。参数除了题目所给的数组candidates以及目标和target之外,为了方便理解,我们算法练习第21天|216.组合总和|||、17.电话号码的字母组合-CSDN博客中的216.组合总和|||的解法一样,添加了目前路径上所遍历的元素的和sum以及开始遍历的位置startIndex。

可能有人有人会有疑问,为什么在17.电话号码的字母组合这一题中没有写startIndex?对于组合问题,什么时候需要startIndex呢?下面给出startIndex的使用场景:

如果是从一个集合中求组合的话,就需要startIndex,例如算法练习第20天|回溯算法 77.组合问题 257. 二叉树的所有路径-CSDN博客中的77.组合问题,以及上述的216.组合总和|||。

如果是多个集合取组合,各个集合之间相互不影响,那么就不用startIndex,例如:上面的17.电话号码的字母组合。

所以本题要求是从同一个集合中求组合,所以要用到startIndex。至于具体怎么用,下面代码中在细说。

回溯函数大致长这样:

  1. vector<vector<int>> result;
  2. vector<int> path;
  3. //回溯第一步,确定回溯函数的参数和返回类型
  4. void backTracking(vector<int>& candidates, int target, int sum, int startIndex){}
  • 回溯函数第二步:确认回溯终止条件 

从下面的树形结构可知, 回溯的终止条件只有 sum== targeth和sum > target两种。如果是sum<target,则说明还有继续寻找并添加元素的可能,所以这种要执行的是单层回溯的逻辑。

所以回溯终止条件长这样:

  1. //回溯第二步,确定回溯终止条件
  2. if(sum == target){
  3. result.push_back(path);
  4. return;
  5. }
  6. else if(sum > target){
  7. return;
  8. }
  •  回溯第三步:确认单层搜索逻辑

能到达这一步,表明sum<target,需要继续进行元素搜索。因此,要做的操作有处理节点(sum加上当前节点,path中记录当前节点),然后就该执行递归了,接着是回溯。代码如下:

  1. //回溯第三步,确认当层回溯逻辑,并回溯
  2. for(int i = startIndex; i < candidates.size(); i++){
  3. sum += candidates[i]; //处理节点
  4. path.push_back(candidates[i]);
  5. backTracking(candidates, target, sum, i); //递归
  6. sum -= candidates[i]; //回溯
  7. path.pop_back();
  8. }

这里我们用到了startIndex,注意我们在递归时的给最后一个参数传的是i,程序初始时传的startIndex肯定为0,所以i最开始为0。但是搜索完最左侧的大分支后,该记录的结果记录下来,该回溯的回溯,i++变成了1,结合下图,为了避免上面所说的【2,3】和【3,2】的组合重复,所以取数时变成了从【3,5】中取数,这就导致startIndex也需要有更新。那么更新为多少时何时呢?观察下图可知当startIndex和此时的i一样时就满足了从【3,5】中取数的条件,于是我们在递归时把i作为了第四个参数的实参。

整体代码:

 整体代码的书写如下:

  1. class Solution {
  2. public:
  3. vector<vector<int>> result;
  4. vector<int> path;
  5. //回溯第一步,确定回溯函数的参数和返回类型
  6. void backTracking(vector<int>& candidates, int target, int sum, int startIndex){
  7. //回溯第二步,确定回溯终止条件
  8. if(sum == target){
  9. result.push_back(path);
  10. return;
  11. }
  12. else if(sum > target){
  13. return;
  14. }
  15. //回溯第三步,确认当层回溯逻辑,并回溯
  16. for(int i = startIndex; i < candidates.size(); i++){
  17. sum += candidates[i]; //处理节点
  18. path.push_back(candidates[i]);
  19. backTracking(candidates, target, sum, i); //递归
  20. sum -= candidates[i]; //回溯
  21. path.pop_back();
  22. }
  23. }
  24. vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
  25. backTracking(candidates,target,0,0);
  26. return result;
  27. }
  28. };

40.组合总和II 

40. 组合总和 II - 力扣(LeetCode)icon-default.png?t=N7T8https://leetcode.cn/problems/combination-sum-ii/description/

题目描述:

给定一个候选人编号的集合 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 <= candidates.length <= 100
  • 1 <= candidates[i] <= 50
  • 1 <= target <= 30

 本人看完解法也不太理解,哎,先放着。。。。

  1. class Solution {
  2. private:
  3. vector<vector<int>> result;
  4. vector<int> path;
  5. void backtracking(vector<int>& candidates, int target, int sum, int startIndex, vector<bool>& used) {
  6. if (sum == target) {
  7. result.push_back(path);
  8. return;
  9. }
  10. for (int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; i++) {
  11. // used[i - 1] == true,说明同一树枝candidates[i - 1]使用过
  12. // used[i - 1] == false,说明同一树层candidates[i - 1]使用过
  13. // 要对同一树层使用过的元素进行跳过
  14. if (i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false) {
  15. continue;
  16. }
  17. sum += candidates[i];
  18. path.push_back(candidates[i]);
  19. used[i] = true;
  20. backtracking(candidates, target, sum, i + 1, used); // 和39.组合总和的区别1,这里是i+1,每个数字在每个组合中只能使用一次
  21. used[i] = false;
  22. sum -= candidates[i];
  23. path.pop_back();
  24. }
  25. }
  26. public:
  27. vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
  28. vector<bool> used(candidates.size(), false);
  29. path.clear();
  30. result.clear();
  31. // 首先把给candidates排序,让其相同的元素都挨在一起。
  32. sort(candidates.begin(), candidates.end());
  33. backtracking(candidates, target, 0, 0, used);
  34. return result;
  35. }
  36. };

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/582072
推荐阅读
相关标签
  

闽ICP备14008679号