【Leetcode】216. Combination Sum III

题目地址：

https://leetcode.com/problems/combination-sum-iii/

从$1\sim 9$取出$k$个数使得和为$n$，每个数最多只能取一次，返回所有的组合。

直接DFS + 剪枝即可。这个是组合型枚举，需要一个变量记录从哪里开始选。递归每一层是在枚举将哪个数加入集合，递归出口是已经枚举出了$k$个数，方案是在递归出口的时候判断。代码如下：

class Solution {
 public:
  vector<vector<int>> combinationSum3(int k, int n) {
    vector<int> v;
    vector<vector<int>> res;
    dfs(1, k, n, v, res);
    return res;
  }

  void dfs(int u, int k, int sum, vector<int>& v, vector<vector<int>>& res) {
    if (v.size() == k) {
      if (!sum) res.push_back(v);
      return;
    }
    
    // 还需要枚举k - v.size()个数，从而k - v.size() <= 9 - i + 1
    for (int i = u; i <= 9 - (k - v.size()) + 1; i++) {
      if (i > sum) return;
      v.push_back(i);
      dfs(i + 1, k, sum - i, v, res);
      v.pop_back();
    }
  }
};

时间复杂度$O(k(\genfrac{}{}{0ex}{}{9}{k}))$，空间$O(k)$，本质上都是常数量级。虽然数字很大，但由于有剪枝，实际上时间复杂度没那么高。