找出所有相加之和为 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]]
回溯法解决
class Solution {
public List<List<Integer>> combinationSum3(int k, int n) {
List<List<Integer>> ans = new ArrayList<>();
backtrack(ans,new ArrayList<>(), k, n, 0, 0, 1);
return ans;
}
public void backtrack(List<List<Integer>> list, List<Integer> tempList, int k, int n, int currentK, int currentN, int num) {
if(currentN > n || currentK > k) {
return ;
}
if(currentK == k && currentN == n) {
//每次产生一个新的list,就直接加到结果里
list.add(new ArrayList<>(tempList));
} else {
for(int i = num; i <= 9; i++) {
tempList.add(i);
backtrack(list, tempList, k, n, currentK + 1, currentN + i, i + 1);
tempList.remove(tempList.size() - 1);
}
}
}
}