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);
            }
        }
    }