在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。
Input
共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
Output
共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
Sample Input
1
8
5
0
Sample Output
1
92
10
dfs有点熟练了,这题的(我觉得)在于对角线的标记,那么刚开始我标记思路错了,后来改了但是TLE,最后打表AC
第一次有问题:
#include<stdio.h>
#include<string.h>
#include<math.h>
int v[100];
int a[100][100];
int ans,n;
void dfs(int x,int y)
{
int i,j,k,p;
if(y>=n)
{
ans++;return;
}
for(i=x;i<n;i++)
{
for(j=0;j<n;j++)
{
if(v[j]==0&&a[k])///这里出事了
{
a[i][j]=1;
v[j]=1;
for(k=0;k<n;k++)
{
for(p=0;p<n;p++)
{
if(abs(j-p)==abs(i-k))
a[k][p]=1; ///没法用上
}
}
dfs(i+1,y+1);
v[j]=0;
for(k=0;k<n;k++)
{
for(p=0;p<n;p++)
{
if(fabs(j-p)==fabs(i-k))
a[k][p]=0;
}
}
}
}
}
return;
}
int main(void)
{
int i,j;
while(scanf("%d",&n)!=EOF)
{
ans=0;memset(v,0,sizeof(v));memset(a,0,sizeof(a));
if(n==0)
break;
dfs(0,0);
printf("%d\n",ans);
}
return 0;
}
问题在于找到前一个皇后的所有对角线点但是没法用上,并且两层循环轻松TLE,因此要换思路
改正:
#include<stdio.h>
#include<string.h>
#include<math.h>
int v[100];
int a[100][100];
int left[100],right[100];
int ans,n;
void dfs(int x,int y)
{
int i,j,k,p;
if(y>=n)
{
ans++;return;
}
for(i=x;i<n;i++)
{
for(j=0;j<n;j++)
{
if(v[j]==0&&left[i-j+n]==0&&right[i+j]==0)
{
a[i][j]=1;
v[j]=left[i-j+n]=right[i+j]=1;
dfs(i+1,y+1);
v[j]=0;
left[i-j+n]=right[i+j]=0;
}
}
}
return;
}
int main(void)
{
int i,j;
while(scanf("%d",&n)!=EOF)
{
ans=0;memset(v,0,sizeof(v));memset(a,0,sizeof(a));memset(left,0,sizeof(left));memset(right,0,sizeof(right));
if(n==0)
break;
dfs(0,0);
printf("%d\n",ans);
}
return 0;
}
优化在哪呢?每一行搜索的时候左右对角线都加入判断,判断能不能放,并且是简单清晰,这个点的对角线点出完了,在下一个点(下一行)的判断中能if到,接着就是模板。
----
当然这个我自己电脑挺快的,但是VJTLE所以最后输出来打表就ok了。