找出所有相加之和为 n
的 k
个数的组合,且满足下列条件:
- 只使用数字1到9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
示例 1:
输入: k = 3, n = 7 输出: [[1,2,4]] 解释: 1 + 2 + 4 = 7 没有其他符合的组合了。
示例 2:
输入: k = 3, n = 9 输出: [[1,2,6], [1,3,5], [2,3,4]] 解释: 1 + 2 + 6 = 9 1 + 3 + 5 = 9 2 + 3 + 4 = 9 没有其他符合的组合了。
示例 3:
输入: k = 4, n = 1 输出: [] 解释: 不存在有效的组合。 在[1,9]范围内使用4个不同的数字,我们可以得到的最小和是1+2+3+4 = 10,因为10 > 1,没有有效的组合。
提示:
2 <= k <= 9
1 <= n <= 60
思路:
本题仍然采用 回溯 + 递归来解决。用 List<List<Integer>> result 来返回最后的组合结果,用 List<Integer> path 来存放每一组符合条件( path 中所有值之和 == n 且 path.size() == k) 的组合。通过 递归实现向下遍历(深度为 k),通过 for 循环实现横向遍历(宽度为 9),此时只能使用 1~9,故 for 循环 的范围是 1~9 。
- 递归不需要返回值。参数为 targetSum,目标值。sum, 当前 path 中所有数之和。k, path 中能容纳值的最大个数。startIndex, 每次递归(向下遍历)开始的位置
- 递归的终止条件是:当 path.size() == k 且 path 中所有值之和 == n 时,将 path 添加进 result 数组中,返回。退出递归后上一层递归中会进行回溯,使 sum 和 path 回退一步,以便进行 for 循环下一个 i 的向下遍历。
- 单层处理的逻辑:利用 for 循环进行横向遍历,对于每一个 i , 将 i 加入 path 里面,并且 sum += i ; 然后向下遍历,调用 backTracking(); 将 targetSum, sum, k ,startIndex 传入,注意此时 startIndex = i + 1。等退出 backTracking() 后,对 sum 和 path 进行回溯,以便对下一个 i 进行处理。
代码:
class Solution {
public List<List<Integer>> result = new ArrayList<>();
public List<Integer> path = new LinkedList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
backTracking(n,0,k,1);
return result;
}
public void backTracking(int targetSum,int sum,int k,int startIndex){
if(path.size()==k){
if(sum==targetSum){//找到一组符合条件的解
result.add(new ArrayList(path));
return;
}
}
//单层处理逻辑:横向遍历
for(int i=startIndex;i<=9;i++){
sum+=i;
path.add(i);
//纵向遍历
backTracking(targetSum,sum,k,i+1);
sum-=i;//回溯
path.removeLast();//回溯
}
}
}
参考:代码随想录