找出所有相加之和为 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]]
本题用到回溯,首先定义一个变量sum代表当前搜索到的数字序列的总和,再定义一个pos变量代表当前遍历的层数,即代表当前取了几个数,从sum=0,pos=0开始递归搜索。每一层都遍历1到9,从中取一个数,sum加上当前取的数,pos也+1,然后递归到下一层,每递归一层就加上当前层遍历到的那个数,同时层数+1。当pos=k时判断sum是否等于n,若等于则将数字序列加入解空间中。若还没有递归到第k层但是总和已经超过了n则返回,若层数超过了k也返回,即回溯操作。代码如下:
public List<List<Integer>> combinationSum3(int k, int n) {
List<List<Integer>> res = new ArrayList<>();
List<Integer> temp = new ArrayList<>();
dfs(res, temp, 0, n, k, 0);
return res;
}
void dfs(List<List<Integer>> res, List<Integer> temp, int pos, int n, int k, int sum) {
if (pos < k && sum > n)
return;
if (pos == k && sum == n) {
res.add(new ArrayList<Integer>(temp));
return;
}
if (pos > k)
return;
for (int i = 1; i <= 9; i++) {
if (temp.contains(i))
return;
temp.add(i);
sum += i;
dfs(res, temp, pos + 1, n, k, sum);
temp.remove(temp.size() - 1);
sum -= i;
}
}