HDUJ 2553 N皇后问题 回溯法

http://acm.hdu.edu.cn/showproblem.php?pid=2553

N皇后问题
Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 8089    Accepted Submission(s): 3610


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<iostream>
#include<cstring>
using namespace std;

int a[11];
int n,ans;

void dfs(int cur,int n)
{
    int i,j;
    if(cur==n)     //递归边界
        ans++;
    else   for(i=0;i<n;i++)
    {
        int ok=1;
        a[cur]=i;       //尝试把第cur行的皇后放在第i列
        for(j=0;j<cur;j++)             //检查是否和前面的皇后冲突
            if(a[cur]==a[j]||cur+a[cur]==j+a[j]||cur-a[cur]==j-a[j])
            {ok=0;break;}
        if(ok)   dfs(cur+1,n);          //如果合法，则继续递归
    }
}

int main()
{
    int b[11];
    for(int i=1;i<=10;i++)
    {
        ans=0;
        dfs(0,i);
        b[i]=ans;
    }
    int m;
    while(cin>>m)
    {
        if(m==0)  break;
        cout<<b[m]<<endl;
    }
    return 0;
}