HDU 2553 N皇后问题 深搜-dfs

题目地址： 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;
}