算法-回溯-组合总和3
1 题目概述
1.1 题目出处
https://leetcode-cn.com/problems/combination-sum-iii/submissions/
1.2 题目描述
找出所有相加之和为 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]]
2 回溯法
2.1 思路
这段转自组合总和-解题思路
回溯法的解体框架是什么呢,解决一个回溯问题,实际上就是一个决策树的遍历过程。一般来说,我们需要解决三个问题:
- 路径:也就是已经做出的选择。
- 选择列表:也就是你当前可以做的选择。
- 结束条件:也就是到达决策树底层,无法再做选择的条件。
其中最关键的点就是:在递归之前做选择,在递归之后撤销选择。
LinkedList result = new LinkedList();
public void backtrack(路径,选择列表){
if(满足结束条件){
result.add(结果);
}
for(选择:选择列表){
做出选择;
backtrack(路径,选择列表);
撤销选择;
}
}
所以我们这里只需要按套路来就行了,不过需要注意一些地方:
- 要将结束条件写在for循环里,才能剪枝,否则会进行不必要递归
- 遍历路径时发现求得的和已经和目标值n相等时,还需要确定是否满足k个数。
2.2 代码
class Solution {
List<List<Integer>> resultList = new LinkedList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
// 最多就是1到9数字相加为45,超过就不合法
if(n > 45){
return resultList;
}
backtrack(k, n, 1, new LinkedList<>(), 0);
return resultList;
}
private void backtrack(int k, int n, int start, List<Integer> tmpList, int tmpSum){
if(k == 0){
// 结束条件1
return;
}
LinkedList<Integer> newList = new LinkedList<>(tmpList);
for(int i = start; i < 10; i++){
// 选择当前数字
int tmpSum2 = tmpSum + i;
if(tmpSum2 > n){
// 结束条件2,剪枝
break;
}
newList.add(i);
if(tmpSum2 == n ){
// 结束条件3,剪枝
if(k == 1){
// 只有这个时候才满足k个数之和为n
resultList.add(newList);
}
break;
}
backtrack(k-1, n, i + 1, newList, tmpSum2);
// 不选当前数字
newList.removeLast();
}
}
}