代码随想录算法训练营第25天

216. 组合总和 III

力扣原题链接

也是采用回溯法，延续上一道组合求和的题的解法思路，只是增加一个路径上的元素求和。

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

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

private void backtracking(int k, int n, int startIndex, int sum) {

    // 剪枝
    if (sum > n) return;

    if (path.size() == k) {
        if (sum == n) {
            ans.add(new ArrayList<>(path));
        }
        return;
    }

    // 剪枝 9 - (k - path.size()) + 1
    for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) {
        path.add(i);
        sum += i;
        backtracking(k, n, i + 1, sum);
        // 回溯
        path.removeLast();
        // 回溯
        sum -= i;
    }

}