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.

Note:

All numbers will be positive integers.

The solution set must not contain duplicate combinations.

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]]

思路

最最開始想的是和3sum一樣外層逐個周遊，内層用雙指針，但是感覺k大的話操作有點麻煩，遞歸應會比較簡單。

後來發現就是一個比較簡單的DFS，

先加入第一個數，1-9九種可能，分别進行DFS，第二個數的選擇從第一個數+1開始，以此類推，pSum記錄目前數字總和；

如果長度到k了直接比較pSum和n值大小判斷是否加入；

如果長度沒到k的時候pSum已經超過了n則不再繼續遞歸。

代碼

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