题目:
找出所有相加之和为 n
的 k
个数的组合,且满足下列条件:
- 只使用数字1到9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
思路:
第一步:确定参数与返回值。参数为n,k,startIndex(遍历起始值),sum(结果列表中数字之和),无返回值
第二步:确定终止条件。当path.size()=k时,若sum=n则将该组合加入结果集,否则直接返回
第三步:确定单层递归逻辑。不剪枝的情况下,for循环从startIndex到9进行遍历搜索组合,将值加入组合,递归,回溯。剪枝优化会有两处,一处是for循环从startIndex到9-(k-path.size())+1,一处是将值加入组合后,若sum>n,则直接回溯。
代码:
private List<List<Integer>> result=new ArrayList<>();
private List<Integer> path=new ArrayList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
backTracking(k,n,1,0);
return result;
}
public void backTracking(int k,int n,int startIndex,int sum){
if(path.size()==k){
if(sum==n) result.add(new ArrayList<>(path));
return;
}
for(int i=startIndex;i<=9-(k-path.size())+1;i++){
sum+=i;
path.add(i);
if(sum>n){//剪枝
sum-=i;//回溯
path.remove(path.size()-1);
return;
}
backTracking(k,n,i+1,sum);//递归
sum-=i;//回溯
path.remove(path.size()-1);
}
}