LeetCode 216. Combination Sum III

LeetCode 216. Combination Sum III (组合总和 III)

题目

链接

https://leetcode.cn/problems/combination-sum-iii/

问题描述

找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：

只使用数字1到9
每个数字 最多使用一次 
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

示例

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

提示

2 <= k <= 9
1 <= n <= 60

思路

同样是回溯法，只需要考虑到剪枝即可。

复杂度分析

时间复杂度 O(n2)
空间复杂度 O(n)

代码

Java

    List<List<Integer>> ans = new ArrayList<>();
    LinkedList<Integer> path = new LinkedList<>();

    public List<List<Integer>> combinationSum3(int k, int n) {
        trace(k, n, 0, 1);
        return ans;
    }

    public void trace(int k, int target, int sum, int start) {
        if (sum > target) {
            return;
        }
        if (sum == target && path.size() == k) {
            ans.add(new ArrayList<>(path));
            return;
        }
        for (int i = start; i <= 9 - (k - path.size()) + 1; i++) {
            path.add(i);
            sum += i;
            trace(k, target, sum, i + 1);
            sum -= i;
            path.removeLast();
        }
    }