题目地址: http://acm.hdu.edu.cn/showproblem.php?pid=2553
N皇后问题是搜索中的经典问题,在N*N的棋盘上放N个皇后,使得每行每列每条对角线上没有两个棋子。
代码里面有详细注释,直接上代码。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
using namespace std;
/*
freopen("input.txt","r",stdin);
freopen("output.txt","w",stdout);
*/
int visit[11];//标记每一列
int maps[11];//maps[i]表示第i行第maps[i]列放置一个皇后
int xh[11];
int ans;//记录有多少种放发
void dfs(int k,int n)//第k行
{
int i,j,flag;
if(k==n+1)//已经放完了n行
{
ans++;
return ;
}
for(i=1;i<=n;i++)//第i列
{
if(visit[i]==0)
{
maps[k]=i;
flag=0;
for(j=1;j<k;j++)//判断第k个位置是否和前k-1个位置在对角线上面
if((maps[k]-maps[j])==(k-j)||(maps[j]-maps[k])==(k-j))
{
flag=1;
break;
}
if(flag==0)//如果没有和前k-1个在对角线上
{
visit[i]=1;
dfs(k+1,n);
visit[i]=0;//这个地方记住要修改过来
}
}
}
}
int main()
{
int i,j,n;
for(i=1;i<=10;i++)
{
memset(visit,0,sizeof(visit));
memset(maps,0,sizeof(maps));
ans=0;
dfs(1,i);//一行一行的放,从第一行放到第i行
xh[i]=ans;
}
while(cin>>n&&n)
{
cout<<xh[n]<<endl;
}
return 520;
}