Leetcode之回溯法专题-216. 组合总和 III(Combination Sum III)
同类题目:
Leetcode之回溯法专题-39. 组合总数(Combination Sum)
Leetcode之回溯法专题-40. 组合总和 II(Combination Sum II)
找出所有相加之和为 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]]
分析:
是之前两题的升级版,之前两题的链接在文章的开头。 给定k和n,求1-9中,取k个数,让他们和为n。 同样的,用回溯法,分为取和不取两个分支,并在之中加上和的判断还有取的个数的判断。
class Solution { List<List<Integer>> ans = new ArrayList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
if(k==0 || n==0) return ans;
dfs(k,n,1,new ArrayList<Integer>(),0);
return ans;
}
public void dfs(int k,int n,int step,ArrayList<Integer> list,int sum){
if(list.size()==k && sum==n){
ans.add(new ArrayList<>(list));
return;
}
if(step>9) return; list.add(step);
dfs(k,n,step+1,list,sum+step);
list.remove(list.size()-1);
dfs(k,n,step+1,list,sum);
}
}