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;
}