LeetCode216组合总和3

题目描述

  找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：只使用数字1到9。每个数字 最多使用一次。返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

解析

  递归加剪枝，搜索长度达到k就可以返回，另外当选择的数之和比n还大的时候就可以不用再搜索了。

public List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> res = new ArrayList<>();
        boolean[] isChosen = new boolean[10];
        List<Integer> current = new ArrayList<>();
        DFS(res, current, isChosen, 0, k, n, 1);
        return res;
    }

    private void DFS(List<List<Integer>> res,
                            List<Integer> current,
                            boolean[] isChosen,
                            int curSum,
                            int k,
                            int n,
                            int start) {

        if(current.size() == k) {
            if(curSum == n) {
                res.add(new ArrayList<>(current));
            }
            return;
        }

        for(int i = start; i <= 9 ; i++) {
            if(!isChosen[i]) {
                current.add(i);
                curSum += i;
                if(curSum <= n) {
                    isChosen[i] = true;
                    DFS(res, current, isChosen, curSum, k, n, i + 1);
                }
                current.remove(current.size() - 1);
                curSum -= i;
                isChosen[i] = false; // 回溯
            }
        }
    }