糖逗 发表于 2020-5-28 10:39:27

C++刷leetcode(216. 组合总和 III)【回溯】

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

说明:

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

输入: k = 3, n = 7
输出: []
示例 2:

输入: k = 3, n = 9
输出: [, , ]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
public:
    void dfs(int k, int n, vector<vector<int> > & res, vector<int>& temp, int cur, int sum){
      if(temp.size() > k) return;
      if(temp.size() == k){
            if(sum == n)res.push_back(temp);
            else if(sum != n) return;
      }
      if(k-temp.size() > 0){
            int store = 9 < (n-sum)/(k-temp.size()) ? 9 : (n-sum)/(k-temp.size());
            for(int i = cur + 1; i <= store; i++){
                temp.push_back(i);
                dfs(k, n, res, temp, i, sum + i);
                temp.pop_back();
            }
      }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
      vector<vector<int> > res;
      vector<int> temp;
      dfs(k, n, res, temp, 0, 0);
      return res;
    }
};
页: [1]
查看完整版本: C++刷leetcode(216. 组合总和 III)【回溯】