1、题目
找出所有相加之和为 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]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2、思路:
有毒吧,这几天每日一题都是组合,组合,组合1,组合2,组合3,吐血。。。DFS一波带走,结束递归的条件-成功:和为n且组合个数为k,失败:和大于n或者组合个数大于k。
需要在每次找到组合后删除最后一个添加元素,才能结束这轮递归;每轮在递归结束必须把当前最后一个元素删除。
因为所求为组合,且数字不能重复,所以设置变量start,从1开始往下走,每次递归start的值为当前所选值+1
忘记把list每次开始时初始化,导致提交时错误两次并且执行时没问题,粗心了
class Solution {
private static List<List<Integer>> list;
private static List<Integer> tem;
public List<List<Integer>> combinationSum3(int k, int n) {
list = new ArrayList<List<Integer>>();
tem = new ArrayList<>();
dfs(k,n,0,0,1);
return list;
}
public void dfs(int k, int n,int sum,int cnt,int start) {
for (int i = start; i < 10; i++) {
if(sum+i > n || tem.size() > k) {
return;
}
tem.add(i);
if(sum+i == n && tem.size() == k) {
List<Integer> tem0 = new ArrayList<>();
tem0.addAll(tem);
list.add(tem0);
tem.remove(tem.size()-1);
return;
}
dfs(k,n,sum+i,cnt+1,i+1);
if(tem.size() > 0) {
tem.remove(tem.size()-1);
}
}
}
}