leecode217 组合优化
找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
回溯三部曲:
(1)确定递归函数参数
需要定义path和result为全局变量
接下来还需要如下参数:
🟢targetSum(int)目标和,也就是题目中的n。
🟢k(int)就是题目中要求k个数的集合。
🟢sum(int)为已经收集的元素的总和,也就是path里元素的总和。
🟢startIndex(int)为下一层for循环搜索的起始位置。
(2)确定终止条件
在上面已经说了,k其实就已经限制树的深度,因为就取k个元素,树再往下深了没有意义。
所以如果path.size() 和 k相等了,就终止。
如果此时path里收集到的元素和(sum) 和targetSum(就是题目描述的n)相同了,就用result收集当前的结果。
class Solution {
List<List<Integer>> result=new ArrayList<>();
LinkedList<Integer> path=new LinkedList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
build(k,n,1,0);
return result;
}
public void build(int k,int n,int startIndex,int sum){
if(sum>n){
return;
}
if(path.size()>k){
return;
}
if(sum==n&&path.size()==k){
result.add(new ArrayList<>(path));
return;
}
for(int i=startIndex;i<=9;i++){
path.add(i);
sum+=i;
build(k,n,i+1,sum);
sum-=i;
path.removeLast();
}
}
}