LeetCode 216. 组合总和 III(Combination Sum III)

Posted FlyingWarrior

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode 216. 组合总和 III(Combination Sum III)相关的知识,希望对你有一定的参考价值。

题目描述

 

找出所有相加之和为 的 个数的组合组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:

  • 所有数字都是正整数。
  • 解集不能包含重复的组合。 

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

 

解题思路

 

回溯算法,从第一个数开始依次添加数字并比较当前数字总和,若相等就添加到结果集合中。

 

代码

 

 1 class Solution {
 2 public:
 3     vector<vector<int>> combinationSum3(int k, int n) {
 4         vector<vector<int>> res;
 5         if(k <= 0 || n <= 0) return res;
 6         vector<int> nums, temp;
 7         for(int i = 0; i < 9; i++)
 8             nums.push_back(i + 1);
 9         combine(nums, 0, 0, k, n, temp, res);
10         return res;
11     }
12     void combine(vector<int> nums, int idx, int sum, int k, int n, vector<int> &temp, vector<vector<int>> &res){
13         if(k == 0 && sum == n)
14             res.push_back(temp);
15         else if(sum < n){
16             for(int i = idx; i < 9; i++)
17                 if(sum + nums[i] <= n){
18                     temp.push_back(nums[i]);
19                     combine(nums, i + 1, sum + nums[i], k - 1, n, temp, res);
20                     temp.pop_back();
21                 }
22         }
23     }
24 };

 

以上是关于LeetCode 216. 组合总和 III(Combination Sum III)的主要内容,如果未能解决你的问题,请参考以下文章

LeetCode 216. 组合总和 III(Combination Sum III)

LeetCode - #216 组合总和 III

Leetcode No.216 组合总和 III

[回溯算法]leetcode216. 组合总和 III(c实现)

Leetcode刷题Python216. 组合总和 III

216. 组合总和 III-----回溯篇4