LeetCode-【数组】- 组合总和 III

版权声明：转载注明出处就可以了。。。 https://blog.csdn.net/zw159357/article/details/82285577

找出所有相加之和为 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;
    }
}