在递归构造中,生成和检查过程可以有机结合起来,从而减少不必要的枚举。
7.4.1 八皇后问题
分析:
当把问题分成若干步骤并递归求解时,如果当前步骤没有合法选择,则函数将送回上一级递归调用,即回溯
#include<stdio.h>
#include<cstring>
using namespace std;
const int maxn=1000;
int vis[3][maxn]; //vis[1][]标记列是否有其他皇后,vis[1][]标记副对角线,vis[2][]标记主对角线;
int C[maxn],tot,n;
void search(int cur)
{
if(cur==n)
{
tot++;
for(int j=0;j<n;j++)
printf("%d ",C[j]);
printf("\n");
}
else for(int i=0;i<n;i++){
if(!vis[0][i]&&!vis[1][cur+i]&&!vis[2][cur-i+n]){
C[cur]=i;
vis[0][i]=vis[1][cur+i]=vis[2][cur-i+n]=1;
search(cur+1);
vis[0][i]=vis[1][cur+i]=vis[2][cur-i+n]=0;
}
}
}
int main()
{
memset(C,0,sizeof C);
memset(vis,0,sizeof vis);
tot=0;
scanf("%d",&n);
search(0);
printf("%d\n",tot);
return 0;
}
7.4.2 其他应用举例
素数环
void dfs(int cur){
if(cur==n&&isp[A[0]+A[n-1]]){
for(int i=0;i<n;i++)
printf("%d ",A[i]);
printf("\n");
}
else for(int i=2;i<=n;i++) //尝试放置每个数i
if(!vis[i]&&isp[i+A[cur-1]]){ //如果i没有用过,并且与前一个数之和为素数
A[cur]=i;
vis[i]=1;
dfs(cur+1);
vis[i]=0;
}
}
困难的串
一种方法是检查所有长度为偶数的子串,分别判断每个子串的前一半和后一半,此方法比较繁琐。
另一种方法只判断当前串的后缀,而非所有子串
void dfs(int cur){
if(cnt++==n){
for(int i=0;i<cur;i++)
printf("%c",'A'+S[i]);
printf("\n");
return 0;
}
for(int i=0;i<L;i++){
S[cur]=i;
int ok=1;
for(int j=1;j*2<=cur+1;i++){ //尝试长度为2*j的后缀
int equal=1;
for(int k=0;k<j;k++) //检查后一半是否等于前一半
if(S[cur-k]!=S[cur-k-j]){
equal=0;
break;
}
if(equal)
{
ok=0;
break;
}
}
if(ok)
if(!dfs(cur+1)) return 0; //递归搜索,如果已经找到解,则直接退出
}
return 1;
}