【Leetcode】39. Combination Sum

题目地址：

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

给定一个数组，都是正整数，再给定一个数$s$，求所有和为$s$的组合。数组中的数可以重复选。

思路是DFS + 剪枝，剪枝方案是，先对candidates排序，这样一来，如果当前的数字和大于了target，就及时结束循环，尝试下一个数字。

枚举的方式有多种。第一种，枚举list下一个数可以放谁。代码如下：

class Solution {
 public:
  vector<vector<int>> combinationSum(vector<int>& a, int t) {
    vector<vector<int>> res;
    sort(a.begin(), a.end());
    vector<int> v;
    dfs(v, 0, 0, t, a, res);
    return res;
  }

  void dfs(vector<int>& v, int s, int pos, int t, vector<int>& a, vector<vector<int>>& res) {
    if (s == t) {
      res.push_back(v);
      return;
    }

    for (int i = pos; i < a.size(); i++) {
      // 后面的数字只能更大，可以直接break
      if (s + a[i] > t) break;
      v.push_back(a[i]);
      dfs(v, s + a[i], i, t, a, res);
      v.pop_back();
    }
  }
};

空间复杂度$O(n)$，时间复杂度$O(k{2}^{n^{\prime }})$，$k$是预期的答案数组的长度，$n^{\prime }$是${\sum }_{i=0}^{n-1}\frac{target}{candidates[i]}$，相当于生成所有组合然后copy一份加入res。

第二种，枚举每个数可以取多少次。代码如下：

class Solution {
 public:
  vector<vector<int>> combinationSum(vector<int>& a, int t) {
    vector<vector<int>> res;
    vector<int> v;
    dfs(v, 0, 0, t, a, res);
    return res;
  }

  void dfs(vector<int>& v, int s, int pos, int t, vector<int>& a,
           vector<vector<int>>& res) {
    if (s == t) {
      res.push_back(v);
      return;
    }

    if (pos == a.size()) return;
	
	// 枚举a[pos]这个数加多少次
    for (int i = 0; s + i * a[pos] <= t; i++) {
      dfs(v, s + i * a[pos], pos + 1, t, a, res);
      v.push_back(a[pos]);
    }
    for (int i = 0; s + i * a[pos] <= t; i++) v.pop_back();
  }
};

时间复杂度和具体题目数据有关。空间$O(n)$。