2n皇后问题 / n皇后问题

题目

点此查看原题

问题描述
　　给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。
输入格式
　　输入的第一行为一个整数n，表示棋盘的大小。
　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。
输出格式
　　输出一个整数，表示总共有多少种放法。
样例输入
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
2
样例输入
4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
0

n皇后问题分析：

• 这个题目其实是在n(8)皇后问题上的进一步，搜索与回溯思路与n皇后问题基本一致。
  先来了解一下n皇后问题。
  首先主函数应该都没问题。然后从棋盘第一行开始Find_answer(1);当找到第n+1行时判断，此时已经查找结束，返回。
  否则遍历前row-1行并判断已经放置的皇后是否与当前位置放置皇后冲突，不冲突时用location[i]记录当前第i行的皇后位置。并继续向下查找第row+1行Find_answer(row+1);

• ?函数涉及到剪枝问题，判断当前位置是否可以放置皇后。columns==location[i]判断之前的行中是否在这一列放置了皇后，abs(row-i)==abs(columns-location[i])判断当前皇后是否与之前放置皇后在同一对角线上~~(如果不理解可以自行画一张图举例试验一下)~~ 。

int valid(int row,int columns){
	for(int i=1;i<=row-1;i++)
		if(columns==location[i]||abs(row-i)==abs(columns-location[i]))
			return 0;
	return 1;}

• 下面放代码，void print()函数可以打印n皇后问题中搜索的结果(已注释)，各位如果想查看结果，可自行粘贴到编译器上去掉注释查看。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>

using namespace std;
int a[15][15];
int ans=0;
int n;
int location[20];
/*void print()
{
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(j==location[i])
				printf("1 ");
			else 
				printf("0 ");}
		printf("\n");}
}*/
int valid(int row,int columns){
	for(int i=1;i<=row-1;i++)
		if(columns==location[i]||abs(row-i)==abs(columns-location[i]))
			return 0;
	return 1;}
void Find_answer(int row){
	if(row==n+1){
		//print();
		ans++;
		//printf("*****************\n");
		return ;}
	for(int i=1;i<=n;i++){
		if(valid(row,i)){
			location[row]=i;
			/*return*/ Find_answer(row+1);}}}//注意：return则只返回其中一种情况，否则输出返回所有情况；
int main() {

	scanf("%d",&n);
	
	Find_answer(1);
	printf("%d",ans);
	return 0;}

2n皇后问题分析

与n皇后思路大体相同，首先搜索黑(或白)皇后的位置，每确定一次黑(或白)皇后的位置，搜索另一色皇后的位置，如果可以找到结果，则ans++，下文代码中用**black_location[]和white_location[]**记录黑白皇后分别放置的位置。注意在搜索白皇后(或黑皇后)时要注意判断此位置是否已经放置另一色皇后，若已放置（i==white_location[row]）则直接跳过当前情况。

#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;

int n,ans;
int a[10][10];
int black_location[10],white_location[10];//不喜欢用全局变量的大?可以写在main（）函数中指针调用

int vaild_white(int row,int conlumn)
{
	for(int i=1;i<row;i++)
		if(conlumn==white_location[i]||abs(row-i)==abs(conlumn-white_location[i]))
			return 0;
	return 1;
}
int vaild_black(int row,int conlumn)
{
	for(int i=1;i<row;i++)
	{
		if(conlumn==black_location[i]||abs(row-i)==abs(conlumn-black_location[i]))
			return 0;
		
	}
		return 1;
}
void Queen_black(int row)
{
	if(row==n+1)
	{
		ans++;
		//return ;
	}
	else 
	{
		for(int i=1;i<=n;i++)
			if(a[row][i]==0||i==white_location[row])
				continue ;
			else if(vaild_black(row,i))
			{
				black_location[row]=i;
				Queen_black(row+1);
			}
	}
}
void Queen_white(int row)
{
	if(row==n+1)
	{
		Queen_black(1);
		return ;
		
	}
	else
	{
		for(int i=1;i<=n;i++)
			if(a[row][i]==0)
				continue;
			else if(vaild_white(row,i))
			{
				white_location[row]=i;
				Queen_white(row+1);
			}
	}

}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			scanf("%d",&a[i][j]);
	Queen_white(1);		
	printf("%d",ans);
	return 0;
			
}