Leetcode 216.组合总和Ⅲ
问题:找出所有相加之和为 n
的 k
个数的组合,且满足下列条件:
- 只使用数字 1 到 9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
算法:
创建二维返回数组 ans ,临时数组 path 。
进入函数 dfs ,边界条件是 d==0 即剩余应选数字个数为 0 ,则可以 return 。如果剩余数字不符合条件,则直跳过,进入下一个循环。
代码:
class Solution {
public:
vector<vector<int>> combinationSum3(int k, int n) {
vector<vector<int>> ans;// 返回数组ans
vector<int> path;// 临时数组path
// 剩余可选择的数字个数i,还需选的数字大小之和t
auto dfs = [&](auto &&dfs,int i,int t){
int d = k - path.size();// 还需选择的数字个数d
if(t < 0 || t > (2*i-d+1)*d/2) return ;// 剩余数字不符合
// 选好了
if(d == 0){
ans.emplace_back(path);
return ;
}
for(int j = i;j >= d;j--){
path.push_back(j);
dfs(dfs,j-1,t-j);
path.pop_back();// 恢复现场
}
};
dfs(dfs,9,n);
return ans;
}
};