C ++中的组合

假设我们有两个整数n和k。我们必须找到1 ... n中k个数字的所有可能组合。因此,如果n = 4且k = 2,则组合将为[[1,2],[1,3],[1,4],[2,3],[2,4],[3,4 ]]

为了解决这个问题,我们将遵循以下步骤-

  • 我们将使用递归函数来解决此问题。该函数solve()采用n,k,临时数组并开始。开始时是1。这就像

  • 如果临时数组的大小= k,则将temp插入res数组,然后返回

  • 对于我:=开始到n

    • 将我插入临时

    • 求解(n,k,temp,i + 1)

    • 从temp的末尾删除一个元素

  • 调用诸如solve(n,k,[])之类的solve函数

  • 返回资源

示例

让我们看下面的实现以更好地理解-

#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]]