HDU 2553 N皇后问题(深搜)

N皇后问题

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

Total Submission(s): 18049    Accepted Submission(s): 8152


[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

<pre name="code" class="cpp">#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
int ap[20][20];
int  ans,n;
int vis[20];
int check(int a,int b)
{
int i,j;
for(i=0; i<a; i++)
for(j=1; j<=n; j++)
if(ap[i][j]==1)
if(abs(j-b)==abs(i-a))//刚开始写成abs((j-b)/(i-a))==1 int
//找了两个小时
return 0;

return 1;
}

void dfs(int k)
{
if(k>n)
{
ans++;
return;
}
for(int j=1; j<=n; j++)
{
if(vis[j]==0)
if(check(k,j))
{
vis[j]=1;
ap[k][j]=1;
dfs(k+1);
vis[j]=0;
ap[k][j]=0;
}

}
}

int main()
{
while(scanf("%d",&n)!=EOF)
{
if(n==0)  break;
ans=0;
memset(vis,0,sizeof(vis));
memset(ap,0,sizeof(ap));
dfs(1);
printf("%d\n",ans);
}
return 0;
}

/*
#include<stdio.h>//打表加回溯
#include<math.h>
#include<algorithm>
#include<string.h>
using namespace std;

int x[15],y[15]= {0};
int vis[20];
int sum,n;
int place(int k)
{
int i;
for(i=1; i<k; i++)
{
if(abs(k-i)==abs(x[k]-x[i])||x[k]==x[i])
//剪枝，即判断是否符合条件来放,i表示皇后所在的行数，x[i]表示所在的列数，
//所以前面那个条件用来判断两个皇后是否在对角线上,后面用来判断是否在同一列上。
//行数不需要判断，因为他们本身的i就代表的是行数
return 0;
}
return 1;
}
void DFS(int a)
{
int i;
if(a>n)
{
sum++;
return;
}
for(i=1; i<=n; i++)
{  if(vis[i]==0)
{x[a]=i;//第a个皇后放的列数
if(place(a))//判断是否能放这步
{vis[i]=1;
DFS(a+1);//能的话进行下一个皇后的放置
vis[i]=0;
}
}
}
}
int main()
{
int i,j,n1;
for(i=1; i<=10; i++)
{   memset(vis,0,sizeof(vis));
n=i;//表示几个皇后
sum=0;//个数每次都要置0
DFS(1);//每次都从第一个皇后开始
y[i]=sum;
}
while(scanf("%d",&n1)==1&&n1)
{
printf("%d\n",y[n1]);
}
return 0;
}
*/