标签：java int chessboard 位置 蓝桥 vip 回退 皇后 type

资源限制

时间限制：1.0s 内存限制：512.0MB

问题描述

　　给定一个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

我的分析：这里的2n皇后（黑白两个）和n皇后问题确实没太大区别，这里就从n皇后问题开始分析。这里它要求棋盘是随机地摆放0和1，所以还是得用个二维数组来表示棋盘，数组元素就存储0或1，黑或白。这个问题按经典算法分为两个部分，如何查找和如何判断。如何查找就是怎样遍历完整个棋盘，找到确定位置；如何判断就是怎样判断当前位置是“正确”的。

我们可以从第一个位置开始检索，从左到右，从上到下。因为要求不能在同一行，所以在当前行找到位置后就可以直接跳转到下一行 查找下一行的位置。因为每一行都是从第一个开始检索并且找到后就立即跳到下一行，所以不用判断同一行是否有相同皇后。我们只需要判断图中的三条红线，即列、左对角线和右对角线。又因为从左到右、从上到下的检索顺序，所以我们只用判断“前半部分”，后面的都还没有检索，所以肯定没有相同元素。如何判断就这样啦，接下来仔细思考如何查找。

用回溯的话，它是检索到了位置不对的地方，就马上回到上一行更变位置。比如我现在找到了第3行，但是每一列的位置都与上面摆好的位置冲突，我们这时就要回退到第2行，假设第2行第4列是之前摆好的，所以我们就是回退到（2，4），然后移到下一个位置也就是（2，5），不能往前移，因为前面的位置都是与上面的位置不符的，所以只能往后移，后面的都是没有检索过的位置。每次回退后，（2，4）这个原位置就要变回1，因为它已经不被我们所用，要及时变回原样。还有就是，当黑白皇后都摆好后，这时count++，函数也要回退，回退摆好的位置就会变为1，因为我们还要找新的摆放位置，所以这是必要的。我最开始还有个疑惑，黑的先还是白的先，黑白位置互换又是一种新的摆法，岂不是最后结果还要乘2？但是回溯的精妙之处解决了这些问题。回退是从后往前回退，第5行的所有可能找完后，它会回退到第4行，又换新的位置摆放，就这样它会回退到最初的第一行第一列这个位置，然后移到（1，2）这个又找新的摆放位置，这样的话，假设我们白的先，（1，1）刚才是被白的占了，现在白的移到（1，2），那么黑的就有机会移到（1，1）这个位置，所以我之前的顾虑黑白位置互换又乘2，是完全不用的。回溯法真的很暴力，它遍历了所有的可能性，以至于保证了我们结果的准确性。这里我们规定2表示白皇后，3表示黑皇后。

import java.util.Scanner;

public class Main {
	//用来存放棋盘
	static int[][] chessboard;
	//棋盘的“阶数”
	static int n;
	//皇后摆放的种类次数
	static int count;
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		chessboard = new int[n][n];
		for (int i = 0; i < n; i++) 
			for (int j = 0; j < n; j++) 
				chessboard[i][j] = in.nextInt();
		
		//默认从第0行开始，先找白皇后
		dfs(0, 2);
		System.out.println(count);
		in.close();
	}
	
	/**
	 * 判断当前位置是否能放
	 * @param x 行坐标
	 * @param y 列坐标
	 * @param type 是白是黑
	 * @return
	 */
	public static boolean check(int x, int y, int type) {
		if (chessboard[x][y] != 1) 
			return false;
		
		//判断当前列上是否有相同皇后
		for (int i = 0; i < x; i++) 
			if (chessboard[i][y] == type) 
				return false;
		
		//判断右对角线上是否有相同元素
		for (int i = x - 1, j = y + 1; i > -1 && j < n; i--, j++) 
			if (chessboard[i][j] == type) 
				return false;
			
		//判断左对角线上是否有相同元素
		for (int i = x - 1, j = y - 1; i > -1 && j > -1; i--, j--) 
			if (chessboard[i][j] == type) 
				return false;
			
		return true;
	}
	
	/**
	 * 摆放位置可以看作是找一条“通路”，当前位置不行就回退用dfs（回溯法）
	 * @param row 棋盘的行数
	 * @param type 2表示白皇后，3表示黑皇后
	 */
	public static void dfs(int row, int type) {
		//最后一行判断
		if (row == n) {
			if (type == 2) //如果白皇后确定好位置就开始摆黑皇后位置
				dfs(0, 3);
			else//若黑皇后摆好次数就加1
				count++;
			return;
		}
		//i表示列数，行数通过递归增长
		for (int i = 0; i < n; i++) 
			if (check(row, i, type)) {
				chessboard[row][i] = type;
				//这一行找好后就直接去下一行找
				dfs(row + 1, type);
				//找完回退到这里，这个位置就变成原样，为了找新的摆放位置
				chessboard[row][i] = 1;
			}
	}

}

标签：java,int,chessboard,位置,蓝桥,vip,回退,皇后,type
来源： https://blog.csdn.net/tastebud/article/details/123035601

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。