找出所有相加之和为 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]]
题解:
- 首先考虑的是分治法,但是由于k不确定,分解成子问题有很多种分法。所以采用递归+回溯
- 1、避免重复,每次递增取数,即上一个取3,下一个从4开始尝试取
- 2、剪枝,越界的部分(元素个数越界、元素和越界)剪除
class Solution {
List<List<Integer>> ret = new ArrayList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
//递归+回溯
List<Integer> list = new ArrayList<>();
combination(list, k, 0, n);
return ret;
}
void combination(List<Integer> list, int cnt, int sum, int target){
if(cnt == list.size() && sum == target){
ret.add(new ArrayList(list));
return ;
}
int last = list.size()==0?0:list.get(list.size()-1);//取出最后一个元素
for(int i=last+1 ; i<=9 ; i++){//防止重复
if(sum+i<=target && list.size()<cnt){
list.add(i);
combination(list, cnt, sum+i, target);
list.remove(list.size()-1);
}
}
}
}