地址 https://leetcode-cn.com/problems/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]]
解答
题目的数据范围不大 就是9个数字的从小到大的组合。
使用dfs深度搜索 将所有数字的组合排列出来,检出符合条件的答案。
注意,不能包含重复组合。那么我们在搜索尝试的时候 需要选择比上一个数字大的数字进行尝试。
class Solution {
public:
vector<vector<int>> ans;
vector<int> numsHash;
bool Check(const vector<int>& v,int n){
int sum = 0;
for(auto& e:v){
sum += e;
}
return sum == n;
}
void dfs(int k ,int n , vector<int>& v){
if(v.size() == k){
if(Check(v,n)){
ans.push_back(v);
}
return ;
}
int startval = 1;
if(!v.empty()){
startval = v.back()+1;
}
for(int i =startval;i<=9;i++){
if(numsHash[i] ==0){
v.push_back(i);
dfs(k,n,v);
v.pop_back();
}
}
return;
}
vector<vector<int>> combinationSum3(int k, int n) {
vector<int> v;
numsHash.resize(10);
dfs(k,n,v);
return ans;
}
};