Leetcode 216 组合总和III

Leetcode 216 组合总和III

题目描述

找出所有相加之和为 n 的 k 个数的组合**。**组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。

说明

• 所有数字都是正整数。
• 解集不能包含重复的组合。

实例1

输入: k = 3, n = 7
输出: [[1,2,4]]

实例2

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

题解

通过dfs，从1开始往下递归每一层，每次+1，退出当层函数时下一层循环+1进入，最后把所有的可能存到列表中。

代码

class Solution {
    public List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> ans = new ArrayList<List<Integer>>();
         for (int i = 1; i <= 9; i++) {
            LinkedList<Integer> list = new LinkedList<Integer>();
            list.add(i);
            getAns(k - 1, n, list, ans);
        }
        return ans;
    }
    
    // list表示当前使用的可能组合
    private void getAns(int k, int n, LinkedList<Integer> list, List<List<Integer>> ans) 		{
        int sum = 0;
        for (int i = 0; i < list.size(); i++)
            sum += list.get(i);
        if (sum == n && k == 0) {
            ans.add((List<Integer>)list.clone());
        }
        // 表示已经超出，移走最后一个数字，重新回到上一层
        if (k <=  0) {
            list.removeLast();
            return;
        }
        // 以最后一个数字+1作为引加入的数字，然后进入新的函数
        int last = list.get(list.size() - 1) + 1;
        for (int i = last; i <= 9; i++) {
            list.add(i);
            getAns(k - 1, n, list, ans);
        }
        // 当正常结束时，移走最后一个数字
        list.removeLast();
    }
}

总结

最后则个答案耗时是2ms内存用了30多M，只能算是中规中矩，在看了优秀的答案以后发现问题是出在代码结构上，上个参考答案:

class Solution {
   	public List<List<Integer>> combinationSum3(int k, int n) {
    List<List<Integer>> combinations = new ArrayList<>();
    List<Integer> path = new ArrayList<>();
    backtracking(k, n, 1, path, combinations);
    return combinations;
}

private void backtracking(int k, int n, int start, List<Integer> tempCombination, List<List<Integer>> combinations) {
    if (k == 0 && n == 0) {
        combinations.add(new ArrayList<>(tempCombination));
        return;
    }
    if (k == 0 || n == 0) {
        return;
    }
    for (int i = start; i <= 9; i++) {
        tempCombination.add(i);
        backtracking(k - 1, n - i, i + 1, tempCombination, combinations);
        tempCombination.remove(tempCombination.size() - 1);
    }
	}
}

---

之前一直习惯用C++来写，现在因为工作用了Java，对Java的传参有点不太适应，简单来说就是四种简单类型和String是类似C++传的值，其他类型都是传的引用了。具体的话可以看下这个知乎答案。Java 到底是值传递还是引用传递。