216. 组合总和 III

题目描述

找出所有相加之和为 nk 个数的组合,且满足下列条件:

  • 只使用数字19
  • 每个数字 最多使用一次

返回 所有可能的有效组合的列表 。

该列表不能包含相同的组合两次

组合可以以任何顺序返回。

示例 1:

输入: k = 3, n = 7

输出: [[1,2,4]]

解释: 1 + 2 + 4 = 7 没有其他符合的组合了。

示例 2:

输入: k = 3, n = 9

输出: [[1,2,6], [1,3,5], [2,3,4]]

解释:

1 + 2 + 6 = 9

1 + 3 + 5 = 9

2 + 3 + 4 = 9

没有其他符合的组合了。

示例 3:

输入: k = 4, n = 1

输出: []

解释:

  • 不存在有效的组合。
  • [1,9]范围内使用4个不同的数字
  • 我们可以得到的最小和是1+2+3+4 = 10
  • 因为10 > 1,没有有效的组合。
提示:
2 <= k <= 9
1 <= n <= 60

题目解析

  • 组合问题
  • 不能重复选择
  • 相当于遍历数组[1, 2, 3, 4, 5, 6, 7, 8, 9]

show code

class Solution {

    private List<List<Integer>> ans;

    private int k;

    private int n;

    private int trackSum = 0;

    public List<List<Integer>> combinationSum3(int k, int n) {
        ans = new ArrayList<>();
        this.k = k;
        this.n = n;

        List<Integer> local = new ArrayList<>();

        // 要求 相加之和 为  n
        // 且要求只能 k 个数相加
        // 不能使用重复数字,不能有重复组合,只能使用数字 1 - 9
        dfs(local, 1);

        return ans;
    }

    private void dfs(List<Integer> local, int num) {
        if(local.size() + (9 - num + 1) < k) {
            return;
        }
        if(trackSum > n || local.size() > k) {
            return;
        }
        if(local.size() == k && trackSum == n) {
            ans.add(new ArrayList<>(local));
            return;
        }

        // 不选择当前数字
        dfs(local, num + 1);

        // 选择当前数字
        local.add(num);
        trackSum += num;
        dfs(local, num + 1);
        // 撤销选择
        local.remove(local.size() - 1);
        trackSum -= num;
    }

}