HRBUST-1249-N皇后问题

题目描述:

在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;
} 

猜你喜欢

转载自blog.csdn.net/weixin_45102820/article/details/113790740