回溯法求组合问题

时间:2022-07-24
本文章向大家介绍回溯法求组合问题,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
#include<iostream>
#include<algorithm>
using namespace std;


bool ok(int get[],int k){
  for(int i=0;i<k;i++)
    if(get[i]>=get[k])
     return false;
return true;
}

void print(int a[],int len){//用与打印 
  for(int i=0;i<len;i++)
    cout<<a[i]<<" ";
      cout<<endl;
}

int main(){
int sum=0;int n,m;
//回溯法从n个数字里面选取m个出来
cout<<"输入总数"<<endl;cin>>n;//读入整数 
int num[n];
cout<<"分别输入这"<<n<<"个数字"<<endl;

for(int i=0;i<n;i++)cin>>num[i];

sort(num,num+n);

cout<<"要取多少个数字组合在一起?"<<endl;
cin>>m;

int get5[m];//需要多少个数字形成组合
 
for(int i=0;i<m;i++) 
  get5[i]=-1;
  
 int k=0;
 int c[m];
for(int i=0;i<m;i++)
    c[i]=0;

     while(k>=0){
     while(c[k]<n){
     get5[k]=num[c[k]++];
if(ok(get5,k)&&k==m-1)//得到一个完整组合 
{
sum++;
}
else if(ok(get5,k)&&k<m-1)k++;//得到部分解,继续往下走 
}
get5[k]==-1;
c[k]=0;
k--;
} 
cout<<"共有"<<sum<<"个组合"<<endl;

}