找出所有相加之和为 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]]
解题思路
结合77题求组合那题,这题的话是加了一个总和的限制。
回溯的道理还是差不多,用path来向深处探索出路,存放元素值,这里要加一个sum来跟targetSum进行比较。当path中元素个数==k,此时判断path中元素和sum是否的等于targetSum,若等于即为一个合格的结果子集,加进结果集res,这个时候不能继续探索了,因为超过k没有意义,所以要结束递归进行return。
for循环是用来在选择集上遍历,即横向遍历,这里给定1~9,所以最大范围到9就行。也可以再结合77题,对 i 再剪枝
还有要注意的一个就是处理过程跟回溯过程一一对应。
作者:jasscical
链接:https://leetcode-cn.com/problems/combination-sum-iii/solution/216zu-he-zong-he-by-jasscical/
class Solution {
public:
vector<vector<int>> res; //存放最终结果集
vector<int> path; //存放向深度探索的路径
void backtracking(int targetSum, int sum, int k, int startIndex){
if(sum > targetSum) return; //剪枝操作
if(path.size() == k){ //如果path中的元素达到题目所给的集合元素限制:k个
if(sum == targetSum){ //并且path中的元素和sum等于给定的targetSum
res.push_back(path); //说明这个结果是合格的,加进结果集
}
return; //path中的元素个数已经等于k,继续下去就会超过k,所以结束此次递归
}
for(int i=startIndex; i<=9; i++){ //选择集合是固定的1~9,所以范围就是到9即可
sum += i; //处理
path.push_back(i);//处理
backtracking(targetSum, sum, k, i+1); //递归
sum -= i; //回溯,与节点处理一一对应
path.pop_back();//回溯,与节点处理一一对应
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(n, 0, k, 1);
return res;
}
};
class Solution {
public:
vector<vector<int>> res; //存放最终结果集
vector<int> path; //存放向深度探索的路径
void backtracking(int targetSum, int sum, int k, int startIndex){
if(sum > targetSum) return; //剪枝操作
if(path.size() == k){ //如果path中的元素达到题目所给的集合元素限制:k个
if(sum == targetSum){ //并且path中的元素和sum等于给定的targetSum
res.push_back(path); //说明这个结果是合格的,加进结果集
}
return; //path中的元素个数已经等于k,继续下去就会超过k,所以结束此次递归
}
for(int i=startIndex; i<=9-(k-path.size())+1; i++){ //剪枝
sum += i; //处理
path.push_back(i);//处理
backtracking(targetSum, sum, k, i+1); //递归
sum -= i; //回溯,与节点处理一一对应
path.pop_back();//回溯,与节点处理一一对应
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(n, 0, k, 1);
return res;
}
};