leetcode216. 组合总和 III

题目

找出所有相加之和为 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个(1-9)的数全排列，如果等于n，则放入结果集，注意一下去重即可。
去重的思路：在dfs方法传入一个参数num，每次全排列的下一个位数都从num+1开始，就可以去除重复的值了。

代码如下

class Solution {
    public List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> res =  new ArrayList<>();
        List<Integer> list = new ArrayList<>();
        dfs(res, list, n,k,0,0);
        return res;
    }

    public void dfs(List<List<Integer>> res, List<Integer> list, int n, int k, int sum, int num) {
        if(sum > n) return;
        if (list.size() == k){
            if(sum == n) {
                res.add(new ArrayList<>(list));
            }
            return;
        }
        for (int i = num+1; i <= 9; i++) {
            list.add(i);
            dfs(res, list, n, k, sum+i, i);
            list.remove(list.size()-1);
        }

    }
}

提交结果

成功
显示详情
执行用时 : 2 ms, 在Combination Sum III的Java提交中击败了67.42% 的用户
内存消耗 : 33.6 MB, 在Combination Sum III的Java提交中击败了29.97% 的用户