##特殊状态的枚举
虽然生成可行空间多数采用深度优先搜索,但在装态空间比较特殊时其实可以简短的实现。比如,c++标准库中提供了next_permutation函数,它可以把n个元素共n!中不同的排列组合生成出来。又或者,通过使用位运算符,可以枚举出从n个元素中取出k个的所有情况,或某个集合的所有子集。
#include<iostream>
#include<algorithm>
#define MAX_N 1000
using namespace std;
bool used[MAX_N];
int perm[MAX_N];
//生成{0,1,2,3,4,……,n-1}种排序
void permutation1(int pos,int n){
if(pos == n){
//在这里编写处理perm的程序
for(int i=0;i<n;i++) cout<<perm[i]<<" ";
cout<<endl;
return;
}
//perm的第pos个位置,究竟使用哪一个元素循环
for(int i=0 ;i<n;i++){
if(!used[i]){
perm[pos]=i;
used[i]=true;
permutation1(pos+1,n);
used[i]=false;
}
}
return;
}
int perm2[MAX_N];
void permutation2(int n){
for(int i=0;i<n;i++){
perm2[i]=i;
}
do{
//此处编写对perm2的操作
for(int i=0;i<n;i++) cout<<perm2[i]<<" ";
cout<<endl;
}while(next_permutation(perm2,perm2+n));
}
int main(){
permutation1(0,5 );
permutation2(5);
return 0;
}
对于1~n-1的全排列,上述代码提供可两种方法:
permutation1方法使用了深度优先搜索算法,该算法为perm数组的每个位置从剩余的元素中挑选一个。
permutation2方法调用了C++ STL提供的头文件下提供的next_permutation方法。
##剪枝
当我们使用穷竭搜索时,程序会把所有可能的情况都列出来,如果有n个数,那么算法是时间复杂度为O(n!)。这样一来,当n较大时,计算会消耗大量时间。
当我们进行深度优先搜索时,到某个状态时已经明确知道在往下搜索不会有解,这时可以跳过,这种行为就称为剪枝。
之前有一个求部分和问题。
例题:部分和问题:
给定整数 a1 、a2 、…… 、an,判断是否可以从中选出若干数,使它们的和恰好为K。
限制条件:1 ≤ n ≤ 20 ;-108≤ ai ≤ 108;-108≤ k≤108;
输入样例:n=4
a={1,2,4,7}
k=13
输出样例:Yes(13 = 2+4+7)
//给出整数a1~an,判断是否可以从中找出若干,使它们的和为k
#include<iostream>
using namespace std;
int n,k, * a; //n为输入数据个数,k为所要的求和,a*为输入的数据数组
void init(){
cin>>n;
a=new int[n];
for(int i=0;i<n;i++){
cin>>a[i];
}
cin>>k;
}
bool dfs(int i,int sum){
if(i==n) return sum==k;
if(dfs(i+1,sum) ) return true;
if(dfs(i+1,sum+a[i]))return true;
return true;
}
int main(){
init();
if(dfs(0,0)) cout<<"Yes";
else cout<<"No";
return 1;
}
该算法其实很简单,如上,dfs只用了五行代码。
假如我们输入的数组长度为5,它们分别为1、2、3、4、5,求它们任意和的总情况有多少种呢?
我们可以这样分析,可以把他作为概率论的问题,从中任意挑几个数,1出现或不出现有两种情况,2出现或不出现有两种情况,……。说以总情况为2^5种。
说以,我们可以以此来画一个二叉树,每层节点代表一个数,左边为不出现,右边为出现:
求部分和https://blog.csdn.net/qq_28120673/article/details/80781806
上述算法中dfs算法代码如下
bool dfs(int i,int sum){
if(i==n) return sum==k;
if(dfs(i+1,sum) ) return true;
if(dfs(i+1,sum+a[i]))return true;
return true;
}
改进该算法:
bool dfs(int i,int sum){
if(i==n) return sum==k;
if(sum>k) return false;//如果sum>k时在往下加一定不会等于k
if(dfs(i+1,sum) ) return true;
if(dfs(i+1,sum+a[i]))return true;
return true;
}