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 瀏覽量

開啟你的 職業

透過完成課程獲得認證

立即上手
廣告