1、题目描述
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。
说明:
- 所有数字都是正整数。
- 解集不能包含重复的组合。
2、示例
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
3、题解
基本思想:递归回溯,深度优先搜索。因为递归回溯所以把vector<int> cur设置为全局变量,另外在dfs中加入n - i >= 0可以减少一些情况继续dfs调用内存资源。
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
vector<vector<int>> res;
vector<int> cur;
vector<vector<int>> combinationSum3(int k, int n) {
//基本思想:递归回溯,深度优先搜索
dfs( k, n, 1);
return res;
}
void dfs(int k, int n, int num)
{
if (k == 0)
{
if (n == 0)
res.push_back(cur);
return;
}
for (int i = num; i < 10 && n - i >= 0; i++)
{
cur.push_back(i);
dfs(k - 1, n - i, i + 1);
cur.pop_back();
}
return;
}
};
int main()
{
Solution solute;
int k = 5, n = 27;
vector<vector<int>> res = solute.combinationSum3(k, n);
for (int i = 0; i < res.size(); i++)
{
for_each(res[i].begin(), res[i].end(), [](const auto v) {cout << v << " "; });
cout << endl;
}
return 0;
}