1. 引言
最近正好刷到暴力枚举相关的专题,在排列组合问题上,一直想寻找一个通用的解决方案,虽然深度优先搜索是比较主流的解决方法,但是无奈太烦,遇到了总是得想老半天。
我们都知道next_permutation函数可以实现数组内元素的全排列,其实稍加变通便可以实现组合问题的解决。
2. 解决方法
next_permutation函数允许待排列的数组元素有重复,但是对于相同的序列不会重复枚举(这里比较难以表述,举个例子)
对于 1 2 3 5 5 这样一个序列,使用next_permutation函数枚举下一个,结果会是 1 2 5 3 5 , 而不是 1 2 3 5 5 。虽然我们认为枚举的下一个应当是第四个元素和第五个元素换了位置的结果,但是由于和上一个序列相同,所以会继续向下枚举。
既然重复的序列不会重复枚举,那么对于组合的问题(比如从n个元素中取k个元素)我们可以构造全为0或者1的序列进行全排列,其中0的个数为n-k个,代表不取,1的个数为k个,代表取,在设置初始状态的时候,让所有的1在后面,确保是枚举的最小的那个状态。下面继续举个例子
对于a b c d e这样一个序列,从中取3个,我们构造这样一个辅助序列 0 0 1 1 1 , 之后使用next_permutation函数进行枚举。枚举结果如下:
0 0 1 1 1
0 1 0 1 1
0 1 1 0 1
0 1 1 1 0
1 0 0 1 1
1 0 1 0 1
1 0 1 1 0
1 1 0 0 1
1 1 0 1 0
1 1 1 0 0
我们看到所有的结果恰好就是代表了所有可能的组合情况,并且不会重复!
3. 例题解决(洛谷p1036)
解决代码如下:
#include<bits/stdc++.h>
using namespace std;
const int MAXN=22;
int n,k;
int num[MAXN];//读入数据存储在这里
int enu[MAXN];//构造的辅助数组,代表了所有的情况
void initenu(){
for(int i=0;i<k;i++){
enu[n-1-i]=1;//根据n,k来初始化辅助数组,确保所有的1都在0后面
}
}
bool isPrime(){
int add=0;
for(int i=0;i<n;i++){
if(enu[i]==1)add+=num[i];
}
if(add==2)return true;
for(int j=2;j<=sqrt(add);j++){
if(add%j==0)return false;
}
return true;
}
int main(){
scanf("%d %d",&n,&k);
for(int i=0;i<n;i++){
scanf("%d",&num[i]);
}
initenu();
int cnt=0;
do{
if(isPrime())cnt++;
}while(next_permutation(enu,enu+n));
printf("%d\n",cnt);
return 0;
}