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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

采用回溯,求和,若是符合要求,则记录。

    public List<List<Integer>> combinationSum3(int k, int n) {

        List<List<Integer>> all = new ArrayList<>();
        List<Integer> list = new ArrayList<>();

        combinationSum3(k, n, 0, 0, all, list);
        return all;
    }

    public static boolean   combinationSum3(int k, int n, int count, int sum, List<List<Integer>> all, List<Integer> list) {
        if (k == count) {
            if (n != sum) {
                return n > sum;
            }
            all.add(new ArrayList<>(list));
        }
        if (n < sum) {
            return false;
        }
        int size = list.size();
        int st = size == 0 ? 1 : list.get(size - 1) + 1;
        for (int i = st; i <= 9; i++) {
            list.add(i);
            boolean b = combinationSum3(k, n, count + 1, sum + i, all, list);
            list.remove(size);
            if (!b) {
                break;
            }
        }
        return true;
    }

posted @ 2021-04-30 15:04  旺仔古李  阅读(47)  评论(0编辑  收藏  举报