[LeetCode 216] Combination Sum III

[LeetCode 216] Combination Sum III

Find all possible combinations of k numbers that add up to a number n, given that only numbers from 1 to 9 can be used and each combination should be a unique set of numbers.

Ensure that numbers within the set are sorted in ascending order.

Example 1:

Input: k = 3, n = 7

Output:

[[1,2,4]]

Example 2:

Input: k = 3, n = 9

Output:

[[1,2,6], [1,3,5], [2,3,4]]

Solution:

DFS, two conditions for basic case, one is sum , second is k number

public List> combinationSum3(int k, int n) {
        List> result = new ArrayList<>();
        List cur = new ArrayList<>();
        if(k<1 || n<1 || k>n) return result;
        dfs(result, cur, 0, k, n, 1);
        return result;
    }
    public void dfs(List> result, List cur, int sum, int k, int n, int level) {
        if(sum == n && k==0) {
            result.add(new ArrayList(cur));
            return;
        } else if(sum> n || k<0) return;
        for(int i=level;i<=9;i++){
            cur.add(i);
            dfs(result, cur, sum+i, k-1, n, i+1);
            cur.remove(cur.size()-1);
        }
    }