【蓝桥杯】 2n皇后问题 (搜索)

50 篇文章 1 订阅
49 篇文章 0 订阅
问题描述
  给定一个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

分析:每行放置黑白两个皇后,他们互不影响。在判断皇后时,黑的和黑的判断,白的和白的判断,因此,可以考虑先将白的皇后放置,然后再放置黑的,如果满足条件,则计数器加1。用2比表示白的皇后,用3表示黑的皇后。

import java.util.Scanner;

/*
 * 2n皇后问题
 * 用2代表白皇后,3代表黑皇后,先将白皇后放置,再放黑皇后
 */
public class No10 {
	public static int[][] map = new int[10][10];
	public static int count=0;
	public static int n =0;
	public static void dfs(int row,int num) {   //第row行放置num皇后
      		if(row ==n+1)                        //如果row==n+1,说明已经完成一轮搜索
     		{
     			if(num==2)                   //如果num=2说明搜索的是白皇后
     			{
     				dfs(1,3);                //搜索黑皇后
     			}else {
     				count++;                //如果num=3,说明黑白皇后都已经按题意放置好了,计数器加1
         			return;
     			}	
     		}
     		for(int col=1;col<=n;col++)      //每一行放置一个皇后,搜索皇后放置的位置
     		{
     			if(isOk(row,col,num))       //判断在col行是否可以放置num皇后
     			{
     				map[row][col] = num;    //放置num皇后
     				dfs(row+1,num);
     				map[row][col] = 1;     //回溯
     			}
     		}
     		
	}
private static boolean isOk(int row, int col, int num) {
	if(map[row][col]==0) return false;
	if(num==3)     //如果num=3,值为0和2的位置都不能放置皇后
	{
		if(map[row][col] ==2)
		{
			return false;
		}
	}
		for(int i=1;i<row;i++)
		{
			for(int j=1;j<=n;j++)
			{
				if(map[i][j]==num)
				{
					if(j==col)      //col行已经放置了皇后
					{
						return false;
					}
					if(row-i==Math.abs(col-j))  //判断对角线上是否放置有皇后
					{
						return false;
					}
				}
			}
		}
	
		return true;
	}
public static void main(String[] args) {
	Scanner in = new Scanner(System.in);
	for(int i=0;i<map.length;i++)
	{
		map[0][i] =-1;
		map[i][0] =-1;
	}
	 n = in.nextInt();
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			map[i][j] = in.nextInt();
		}
	}
	dfs(1,2);
	System.out.println(count);
}

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值