N皇后问题(HDU-2553)

Problem Description

在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。
你的任务是，对于给定的N，求出有多少种合法的放置方法。

 

Input

共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

 

Output

共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

 

Sample Input

  
  

   
   1
8
5
0
  
  

 

Sample Output

  
  

   
   1
92
10
  
  

 

代码如下：

#include<bits/stdc++.h>
using namespace std;
int m[10],vis[20][20],n,cnt;
int ok(int a,int b)
{
    for(int i=0;i<n;i++)if(vis[a][i]||vis[i][b])return 0;
    for(int i=a-b>=0?a-b:0;i<a;i++)if(vis[i][i-a+b])return 0;
    for(int i=a+b-n>=0?a+b-n:0;i<a;i++)if(vis[i][a+b-i]) return 0;
    return 1;
}
void dfs(int x)
{
    if(x==n)
        cnt++;
    for(int i=0;i<n;i++)
        if(ok(x,i))
        {
            vis[x][i]=1;
            dfs(x+1);
            vis[x][i]=0;
        }
}
int main()
{
    for(n=1;n<=10;n++)
    {
        cnt=0;
        dfs(0);
        m[n]=cnt;
    }
    while(cin>>n&&n!=0)
        cout<<m[n]<<endl;
    return 0;
}