LeetCode 216. 组合总和 III(Combination Sum III)
题目描述
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 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 };