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, ],]
廣告
資料結構
網路
RDBMS
作業系統
Java
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP