hdu 2553 N皇后问题（回溯）

hdu 2553 N皇后问题（回溯）

N皇后问题

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
  
  

 

Author

cgf

 

皇后问题早就做过了，这次却花了一下午，我也是无语！看了半天，然后查了一下题解，再想想，终于想明白是标记列和斜面重叠了！多的没什么好说

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
bool vis[3][22];//开始就想用一维的vis[44]来标记的！就是死也做不出！
int n, sum, dp[12];
void dfs(int cur)
{
    if(cur == n) sum ++; 
    else for(int i = 0;i < n; i ++) 
       if(!vis[0][i] && !vis[1][i+cur] && !vis[2][cur-i+n])
        {
            vis[0][i] = vis[1][i+cur] = vis[2][cur-i+n] = 1;
            dfs(cur+1);
            vis[0][i] = vis[1][i+cur] = vis[2][cur-i+n] = 0;
        }
}
int main()
{
    for(n = 1; n < 11; n ++)
    {
        sum = 0;
        dfs(0);
        dp[n] = sum;
    }
    while(scanf("%d", &n), n) printf("%d\n", dp[n]);
    return 0;
}