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, ],]
廣告