当前位置:   article > 正文

C++~LeetCode216-组合总和 III(回溯)_c++ 组合总和iii

c++ 组合总和iii

216. 组合总和 III

找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:

  • 只使用数字1到9
  • 每个数字 最多使用一次 

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。

力扣

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
解释:
1 + 2 + 6 = 9
1 + 3 + 5 = 9
2 + 3 + 4 = 9
没有其他符合的组合了。

示例 3:

输入: k = 4, n = 1
输出: []
解释: 不存在有效的组合。
在[1,9]范围内使用4个不同的数字,我们可以得到的最小和是1+2+3+4 = 10,因为10 > 1,没有有效的组合。

提示:

  • 2 <= k <= 9
  • 1 <= n <= 60

思路:采用回溯枚举的方式求解。

        本题让我们求组合,组合是无序的,所以1,2    2,1算作是同一个组合,而且题目要求每个数只能使用一次。因此,我们在枚举的时候,如果第一个数是2,那么下一个数我们应该从3开始,而不是1或者2。

首先我们先写好回溯求解的大致模板

  1. void dfs() {
  2. if (终止条件) {
  3. //终止访问
  4. }
  5. for (本层集合中的每个与元素) {
  6. //处理元素
  7. dfs(递归调用对下一层的结点求解)
  8. //回溯,找另一条路径
  9. }
  10. }

        接着我们要确定传入的参数,题目给出的n和k以外,我们还需要一个nowSum来表示当前结果集中的和是多少,startNum表示现在从第几个数开始往后访问,因为组合是无序的,这里startNum的目的就是为了防止重复的组合出现。实际上我们还需要两个参数,一个是当前的结果集,一个是总的结果集,这里我定义成了全局变量,如果作为参数传入则需要加上&,切记。

        然后就是确定终止条件,如果当前结果集的数量是k并且和为n,那就将当前结果集加入总的结果集,否则如果出现和>n,结果集size>k,直接返回,相当于顺便做了剪枝

        我们把整个问题的求解看成一颗树,树上的每条路径都是问题的一个可能的解,接着我们就是用for循环遍历当前层的集合元素,也就是当前层的结点,首先从startNum这个数开始,加入结果集,然后递归的去子树中寻找合适的结果集,找完以后返回。去当前层的另一条路径上去找合适的解。

最终代码:

  1. class Solution {
  2. public:
  3. vector<vector<int>> ans;
  4. vector<int> path;
  5. void dfs(int n, int k, int nowSum, int startNum) {
  6. if (nowSum == n && k == path.size()) {
  7. ans.push_back(path);
  8. return;
  9. } else if (nowSum > n) {
  10. return;
  11. } else if (k < path.size()) {
  12. return;
  13. }
  14. for (int i = startNum; i <= 9; i++) {
  15. path.push_back(i);
  16. dfs(n, k, nowSum + i, i + 1);
  17. path.pop_back();
  18. }
  19. }
  20. vector<vector<int>> combinationSum3(int k, int n) {
  21. dfs(n, k, 0, 1);
  22. return ans;
  23. }
  24. };

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

闽ICP备14008679号