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]]
广告