题目描述:
在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。
Input:
共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
Output:
共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
思路:
看到题目,显然是DFS搜索,如何搜索呢?这里的N是N行N列,N个皇后,任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。这样的话,我们可以从第一行开始放皇后,然后一行一行放,在放每一行时,判断是否可以放到某一列,如果可以,就下一行,一直到第n+1个就结束搜索,下边看代码。
我们用map[x]表示这一个皇后,放在第x行第map[x]列,check(x)函数用来判断该皇后时候可以放在这一行这一列,可以的话就返回1,不可以就返回0。
需要注意的是,这里需要先打表,否则会超时。
int check(int x)//检查x行map[x]列是否可以放
{
for(int i=1;i<x;i++)
{
if(map[i]==map[x]||abs(x-i)==abs(map[x]-map[i]))//列相等或者对角线成45度,返回0
return 0;
}
return 1;
}
完整代码:
#include <iostream>
#include <cmath>
using namespace std;
int n,sum,map[15],a[15];
int check(int x)//检查x行map[x]列是否可以放
{
for(int i=1;i<x;i++)
{
if(map[i]==map[x]||abs(x-i)==abs(map[x]-map[i]))//列相等或者对角线成45度,返回0
return 0;
}
return 1;
}
void dfs(int x)//第x行
{
if(x>n)
{
sum++;
return;
}
for(int i=1;i<=n;i++)
{
map[x]=i;
if(check(x))//可以放
dfs(x+1);
}
}
int main()
{
for(int i=1;i<=10;i++)
{
sum=0;
n=i;
dfs(1);
a[i]=sum;
}
while(cin>>n&&n)
{
cout<<a[n]<<endl;
}
return 0;
}