题目

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明

  • 所有数字都是正整数。
  • 解集不能包含重复的组合。

代码

    // 2020-09-09 11:55:20
    public List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> res = new ArrayList<>();
        int[] nums = {1, 2, 3, 4, 5, 6, 7, 8, 9};
        dfs(nums, 0, n, k, new ArrayDeque<>(), res);
        return res;
    }

    private void dfs(int[] nums, int pos, int target, int k, ArrayDeque<Integer> deque, List<List<Integer>> res) {
        if (target == 0 && deque.size() == k) {
            res.add(new ArrayList<>(deque));
            return;
        }
        if (deque.size() == k) {
            return;
        }
        for (int i = pos; i < nums.length; i++) {
            if (target - nums[i] < 0) {
                break;
            }
            deque.add(nums[i]);
            dfs(nums, i + 1, target - nums[i], k, deque, res);
            deque.removeLast();
        }
    }