思路:仿照【回溯法】力扣77题 组合 记录_sheep_fei的博客-优快云博客
进行回溯(穷举),形成树形结构穷举
修改判断条件:维持一个sum记录当前所选元素的累计和与n比较
优化:剪枝操作 1 元素个数不够
2 后面最大(9)加起来不够大
3 当前sum已经比n大
class Solution {
public:
vector<vector<int>> results;//结果
vector<int> path;//栈
int sum;
vector<vector<int>> combinationSum3(int k, int n) {
dfs(1, 9, k, n);
return results;
}
void dfs(int start, int end, int k, int n)
{
if(end - start + 1 < k - path.size()) return;//剪枝,后面元素个数不够无需穷举
if(sum + 9*k < n) return;//剪枝,如果后面加起来一定不够大
if(sum > n) return;//剪枝,如果已经大了
if(path.size() == k && sum == n) {//path已经组成k个数,加入results
results.push_back(path);
return;
}
for(int i = start; i <= end; ++i)
{
path.push_back(i);//放入i
sum += i;
dfs(++start, end, k, n);//进入下一层子集
int top = path.back();
path.pop_back();//回退,选择其他路径
sum -= top;
}
}
};