134 字 少于 1 分钟阅读

难度中等326收藏分享切换为英文接收动态反馈

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

说明:

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

示例 1:

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

示例 2:

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

思路

回溯

解法

class Solution {

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

    public List<List<Integer>> combinationSum3(int k, int n) {
        if (k == 0) {
            return result;
        }
        backtracking(1, k, n);
        return result;
    }

    public void backtracking(int start, int k, int n) {
        if (k == 0) {
            if (n == 0) {
                result.add(new ArrayList<>(list));
            }
            return;
        }
        for (int i = start; i <= 9 - k + 1 && n - i >= 0; i++) {
            list.add(i);
            backtracking(i + 1, k - 1, n - i);
            list.remove(list.size() - 1);
        }
    }
}