hdu-2553 N皇后问题

N皇后问题

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 13272    Accepted Submission(s): 6005


[align=left]Problem Description[/align]
在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。

你的任务是，对于给定的N，求出有多少种合法的放置方法。

 

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

[align=left]Output[/align]
共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。
 

[align=left]Sample Input[/align]

1
8
5
0

 

[align=left]Sample Output[/align]

1
92
10

 

[align=left]Author[/align]

# include<stdio.h>
# include<string.h>
# include<algorithm>
# define max 100 + 10
using namespace std;
int mark[3][max];
int dp[max], sum;

void dfs(int num,int n)
{
if(num == n + 1) sum++;
else
{
for(int i = 1; i <= n; i++)
{
if(!mark[0][i] && !mark[1][i+num] && !mark[2][num-i+n])
{
mark[0][i] = mark[1][i+num] = mark[2][num-i+n] = 1;
dfs(num + 1, n);
mark[0][i] = mark[1][i+num] = mark[2][num-i+n] = 0;
}
}
}
}

int main()
{
int n;
memset(dp, 0 ,sizeof(dp));
for(int i = 1; i <= 10; i++)
{
sum = 0;
memset(mark, 0, sizeof(mark));
dfs(1,i);
dp[i] = sum;
}
while(scanf("%d",&n),n)
{
printf("%d\n",dp
);
}
return 0;
}