LeetCode——深度优先搜索应用(组合总数3)

找出所有相加之和为 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]]

private List<List<Integer>> ans = new ArrayList<>();
    private int[] paths = new int[100];

    // 1-7的数字   index表示第几层
    private void robot(int index, int k, int n, int j) {
        if (index >= k) {
            List<Integer> temp = new ArrayList<>();
            int sum = 0;
            for(int i = 0; i < k; i++) {
                sum += paths[i];
                temp.add(paths[i]);
            }
            if (sum == n) {
                ans.add(temp);
            }
            return;
        }
        for(int i = j; i <= 9; i++) {
            paths[index] = i;
            robot(index + 1, k, n, i+1);
        }

    }

    public List<List<Integer>> combinationSum3(int k, int n) {
        ans.clear();
        robot(0, k, n, 1);
        return ans;

    }

猜你喜欢

转载自blog.csdn.net/reuxfhc/article/details/80472154