C++ 中的組合
假設我們有 2 個整數 n 和 k。我們必須從 1 ... n 中找到 k 個數字的所有可能組合。因此,如果 n = 4 且 k = 2,則組合將為 [[1,2],[1,3],[1,4],[2,3],[2,4],[3,4]]
要解決此問題,我們將按照以下步驟進行操作 −
- 我們將使用遞迴函式來解決此問題。函式 solve() 接收 n、k、臨時陣列和 start。start 初始值為 1。這將起到這樣的作用
- 如果臨時陣列的大小 = k,則將臨時陣列插入 res 陣列,然後返回
- 對於 i := start 到 n,
- 將 i 插入臨時陣列
- solve(n, k, temp, i + 1)
- 從臨時陣列的末尾移除一個元素
- 像 solve(n, k, []) 這樣呼叫 solve 函式
- 返回 res
示例
讓我們看看下面的實現來更好地理解 −
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<vector<int> > 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 n, int k, vector <int> temp, int start = 1){
if(temp.size() == k){
res.push_back(temp);
return;
}
for(int i = start; i <= n; i++){
temp.push_back(i);
solve(n, k, temp, i + 1);
temp.pop_back();
}
}
vector<vector<int> > combine(int n, int k) {
res.clear();
vector <int> temp;
solve(n ,k, temp);
return res;
}
};
main(){
Solution ob;
print_vector(ob.combine(5,3));
}輸入
5 3
輸出
[[1,2,3],[1,2,4],[1,2,5],[1,3,4],[1,3,5],[1,4,5],[2,3,4],[2,3,5],[2,4,5],[3,4,5]]
廣告
資料結構
網路
RDBMS
作業系統
Java
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP