C++ 中的组合求和 IV


假设我们不得不生成所有可能的 k 个数字组合,这些数字加起来等于 n,且只能使用数字 1 至 9。每个组合都应为一组唯一的数字。所有数字都应为正数,且解中不得包含重复的组合。那么,如果 k = 3,n = 9,可能的组合是 [[1,2,6],[1,3,5],[2,3,4]]

要解决这个问题,我们将按照以下步骤操作 -

  • 假设我们将使用一个名为 solve 的方法来解决它。这将是一个递归方法,它将把 k、n、temp 阵列和 start 作为参数。start 最初为 1
  • if n = 0
    • if temp 的大小 = k,则将 temp 插入 res 并返回
    • 对于 i := start 到 9 和 n 的最小值
      • 将 i 插入 temp
      • solve(k, n – i, temp, i + 1)
      • 删除 temp 中的最后一个元素
  • 主要方法如下
  • 创建一个名为 temp 的空向量
  • solve(k, n, temp)
  • 返回 res

我们来看看以下实现,以便更好地理解 -

示例

 在线演示

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<vector<auto> > v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << "[";
      for(int j = 0; j <v[i].size(); j++){
         cout << v[i][j] << ", ";
      }
      cout << "],";
   }
   cout << "]"<<endl;
}
class Solution {
   public:
   vector < vector <int> > res;
   void solve(int k, int n, vector <int> temp, int start = 1){
      if(n == 0){
         if(temp.size() == k){
            res.push_back(temp);
         }
         return;
      }
      for(int i = start ; i <= min(9, n); i++){
         temp.push_back(i);
         solve(k, n - i, temp, i + 1);
         temp.pop_back();
      }
   }
   vector<vector<int>> combinationSum3(int k, int n) {
      res.clear();
      vector <int> temp;
      solve(k, n, temp);
      return res;
   }
};
main(){
   Solution ob;
   print_vector(ob.combinationSum3(2, 9));
}

输入

3
9

输出

[[1, 8, ],[2, 7, ],[3, 6, ],[4, 5, ],]

更新于: 04-May-2020

108 浏览量

开启你的 职业

通过完成课程获得认证

立即上手
广告