N皇后问题Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 37380 Accepted Submission(s): 15887 Problem Description 在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
Input 共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
Output 共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
Sample Input 1 8 5 0
Sample Output 1 92 10
|
思路:
1) 算法开始, 初始化棋盘,初始行设为第一行,当前列设为第一列
2) 在当前行,当前列的位置上判断是否满足条件(即保证经过这一点的行,列与斜线上都没有两个皇后),若不满足,跳到第4步
3) 在当前位置上满足条件的情形:
在当前位置放一个皇后,若当前行是最后一行,记录一个解;
若当前行不是最后一行,移动当前行到为下一行, 当前列设为当前行的第一个待测位置;
若当前行是最后一行,当前列不是最后一列,移动当前列到下一列;
若当前行是最后一行,当前列是最后一列,回溯,即清空当前行及以下各行的棋盘,然后,移动当前行到上一行,当前列设为当前行的下一个待测位置;(重新寻找另一个解)
以上返回到第2步
4) 在当前位置上不满足条件的情形:
若当前列不是最后一列,移动当前列设为下一列,返回到第2步;
若当前列是最后一列了,回溯,即,若当前行已经是第一行了,算法退出,否则,清空当前行及以下各行的棋盘,然后,当前行设为上一行,当前列设为当前行的下一个待测位置,返回到第2步;
注意需要先打表,不然会TLE
代码:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#define inf 0x3f3f3f3f
using namespace std;
int q[11]; //每个皇后所在的列
int sum;
int res[11];
bool find(int row,int col) //行、列
{
int k=1;
while(k<row) //k=1~k-1是已经放置了皇后的行
{
if(q[k]==col || abs(k-row)==abs(q[k]-col)) //检查是否在同一列,检查是否在对角线上
return false;
k++;
}
return true;
}
void dfs(int row,int n)
{
if(row>n){
sum++;
}
else
{
for(int i=1;i<=n;i++) //搜索第row行的每一列
{
if(find(row,i)) //查询在第row行第i列放置皇后是否合法
{
q[row]=i; //放置皇后
dfs(row+1,n); //搜索下一行
}
}
}
}
int main()
{
int n;
//打表
for(n=1;n<=10;n++)
{
sum=0;
dfs(1,n); //每次都从1开始放置,将所有情况的结果打出来
res[n]=sum;
}
while(~scanf("%d",&n) && n)
{
// sum=0;
// dfs(1,n); //初始位置第一行
printf("%d\n",res[n]);
}
return 0;
}
debug过程:
一开始n=10的时候输出一直是5,而正确的结果应该是724
仔细检查发现 q[10]数组下标越界了,赶紧改成q[11]