LeetCode-216. 组合总和 III-Java-medium

题目链接

法一（回溯）

public class Solution216 {

    private List<List<Integer>> ans; // 符合条件的结果集
    private List<Integer> path; // 已经收集的元素
    private int sum; // path里元素的总和

    /**
     * 无参构造函数
     */
    public Solution216() {
        this.ans = new LinkedList<>();
        this.path = new LinkedList<>();
        this.sum = 0;
    }

    /**
     * 回溯
     *
     * @param n
     * @param k
     * @param start
     */
    private void backtracking(int k, int n, int start) {
        if (sum > n) { // 剪枝
            return;
        }
        if (path.size() == k) {
            if (sum == n) {
                ans.add(new LinkedList<>(path));
            }
            return;
        }
        for (int i = start; i <= 10 - (k - path.size()); i++) { // 剪枝（横向遍历），剩余个数凑不齐k个的i直接跳过
            sum += i;
            path.add(i); // 处理节点
            backtracking(k, n, i + 1); // 通过不断调用自己一直往深处遍历（纵向遍历）
            sum -= i;
            path.remove(path.size() - 1); // 回溯，撤销处理的节点
        }
    }

    /**
     * 回溯法
     * （1）9（只使用数字1到9）相当于树的宽度，k相当于树的深度
     * （2）每次搜索到了叶子节点，就找到了一个结果
     * （3）只需要把达到叶子节点的结果收集起来，就可以找到所有相加之和为n的k个数的组合
     *
     * @param k
     * @param n
     * @return
     */
    public List<List<Integer>> combinationSum3(int k, int n) {
        backtracking(k, n, 1);
        return ans;
    }
    
}

本地测试

        /**
         * 216. 组合总和 III
         */
        lay.showTitle(216);
        Solution216 sol216 = new Solution216();
        int k216 = 3, n216 = 9;
        List<List<Integer>> ans216 = sol216.combinationSum3(k216, n216);
        arrayOpt.showIntLists(ans216);