【算法入门经典】7.4回溯法【八皇后问题】

#include<stdio.h>
#include<string.h>
int ans,n, map[20];
void search(int cur)
{
int i,j,flag;
if(cur == n+1)//递归边界
ans ++;
else
{
for(i = 1; i <= n; i ++)
{
map[cur] = i;//把第cur行的皇后放在第i列
flag = 1;
for(j = 1; j < cur; j ++)
{//判断是否和前面的皇后冲突
if(map[cur] == map[j]||cur-map[cur]==j-map[j]||cur+map[cur]==j+map[j])
{
flag = 0;break;
}
}
if(flag)//如果合法，则继续递归
search(cur+1);
}
}
}
int main()
{
while(scanf("%d",&n),n!=0)
{
ans = 0;
memset(map,0,sizeof(map));
search(1);
printf("%d\n",ans);
}
return 0;
}

#include<stdio.h>
#include<string.h>
#define N 100
int ans, n ,vis[3][100];
void dfs(int cur)
{
int i;
if( cur == n+1)
ans ++;
else
{
for(i = 1; i <= n; i ++)
{
if(!vis[0][i]&&!vis[1][cur+i]&&!vis[2][cur-i+n])//利用二维数组直接判断
{//该列没有放过棋子&&主对角线没有放过棋子&&副对角线没有放过棋子
vis[0][i] = vis[1][cur+i] = vis[2][cur-i+n] = 1;//修改全局变量
dfs(cur+1);
vis[0][i] = vis[1][cur+i] = vis[2][cur-i+n] = 0;//回溯
}
}
}
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
ans = 0;
memset(vis,0,sizeof(vis));
dfs(1);
printf("%d\n",ans);
}
return 0;
}