C++组合数 II


假设我们有一组候选数字(所有元素都是唯一的)和一个目标数字。我们必须找到候选数字中所有唯一的组合,其中候选数字之和等于给定的目标值。同一个数字不会从候选数字中选择超过一次。所以如果元素是[2,3,6,7,8],目标值是8,那么可能的输出将是[[2,6],[8]]

让我们看看步骤:

  • 我们将以递归的方式解决这个问题。递归函数名为solve()。它接受索引、数组a、整数b和另一个数组temp。solve方法的工作方式如下:
  • 定义空数组res
  • 如果b = 0,则将temp插入res并返回
  • 如果index = a的大小,则返回
  • 如果b < 0,则返回
  • 对数组a进行排序
  • 对于i的范围从index到a的大小-1
    • 如果i > index并且a[i] = a[i – 1],则继续
    • 将a[i]插入temp
    • solve(i + 1, a, b – a[i], temp)
    • 删除temp中的最后一个元素
  • 通过传递index = 0、数组a和目标b以及另一个数组temp来调用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 idx, vector <int> &a, int b, vector <int> temp){
      if(b == 0){
         res.push_back(temp);
         return;
      }
      if(idx == a.size())return;
      if(b < 0)return;
      sort(a.begin(), a.end());
      for(int i = idx; i < a.size(); i++){
         if(i > idx && a[i] == a[i-1])continue;
         temp.push_back(a[i]);
         solve(i + 1, a, b - a[i], temp);
         temp.pop_back();
      }
   }
   vector<vector<int> > combinationSum2(vector<int> &a, int b) {
      res.clear();
      vector <int> temp;
      solve(0, a, b, temp);
      return res;
   }
};
main(){
   Solution ob;
   vector<int> v = {2,3,6,7,8};
   print_vector(ob.combinationSum2(v, 10)) ;
}

输入

[2,3,6,7,8]
8

输出

[[2, 8, ],[3, 7, ],]

更新于:2020年5月4日

439 次浏览

开启你的职业生涯

通过完成课程获得认证

开始学习
广告
© . All rights reserved.