找出所有相加之和为 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]]
题解:这个题目和求一个数组的子集的题目基本相似,不过是加了两个递归条件,一是子集和为n,二是子集长度为k,有这个思路很容易写出代码,这里需要注意,如果直接用这两个条件判断,会有很多不必要的计算,一是子集求和大于n,二是虽然子集和小于n,但子集的长度已经大于k,这两种情况都可以直接排除掉。
class Solution {
public static void dfs(List<List<Integer>> res,List<Integer> tmp,int[] nums,int s,int n,int k){
if(n<0)
return;
if(tmp.size()>k)
return;
if(n==0&&tmp.size()==k){
res.add(new ArrayList<>(tmp));
return;
}
for(int i=s;i<nums.length;i++){
tmp.add(nums[i]);
dfs(res,tmp,nums,i+1,n-nums[i],k);
tmp.remove(tmp.size()-1);
}
}
public List<List<Integer>> combinationSum3(int k, int n) {
List<List<Integer>> res=new ArrayList<>();
int[] nums=new int[9];
for(int i=0;i<9;i++)
nums[i]=i+1;
dfs(res,new ArrayList<>(),nums,0,n,k);
return res;
}
}