蓝桥杯【2n皇后问题】Java

问题描述

    　给定一个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 x n 的棋盘摆放n个黑皇后和n个白皇后，且同肤色皇后不能同行同列同对角线，则每行必定为一黑一白。可令2代表黑皇后，3代表白皇后，我们先将黑皇后安顿好再安顿白皇后。首先挨行检索有无皇后容身之处，前提要满足该位置为1，其次判断前几行放置的黑皇后是否与当前位置不同列且不同对角线，若以上条件都满足则将皇后放到当前位置（将当前位置设置为2），若不满足则检索该行下一位置，倘若整行都没有合适位置，则回溯至上一行移动黑皇后的位置，重复以上步骤，对于白皇后的操作亦是如此。另外，在一种方案检索完毕后，要记得还原棋盘，以便下一方案可行性的测试。

代码实现

import java.util.Scanner;

public class Main {
   
	static int n, sum = 0;// n:棋盘边长(皇后数量),sum:不同摆法
	static int[][] chess;// 棋盘

	public static void main(String[] args) {
   
		Scanner sc = new Scanner(System