原题链接:[216. 组合总和 III(https://leetcode.cn/problems/combination-sum-iii/submissions/483547922/)
思路:
跟77.组合差不多,此时只需要遍历全部结点并且进行回溯操作就行
终止条件方面,依旧是path.size() == k 时,就终止
但是需要在进行一次判断,如果sum等于题目给定的targetSum,则代表是需要的值 将path给push_back进result中即可
回溯时要注意,sum要进行 -i,并把path给pop_back
全代码:
class Solution {
public:
vector<vector<int>> result; // 存放结果集
vector<int> path; // 符合条件的结果
// targetSum:目标和,也就是题目中的n。
// k:题目中要求k个数的集合。
// sum:已经收集的元素的总和,也就是path里元素的总和。
// startIndex:下一层for循环搜索的起始位置。
void backtracking(int targetSum, int k, int sum, int startIndex) {
if(path.size() == k){//
if(sum == targetSum)
{
result.push_back(path);
}
return ;
}
for(int i = startIndex; i <= 9; i++)
{
sum += i; //计算总和
path.push_back(i);//往path里面加
backtracking(targetSum, k , sum , i + 1);
sum -= i; //回溯操作
path.pop_back(); //回溯
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(n, k, 0, 1);
return result;
}
};