蓝桥杯VIP 基础练习 2n皇后问题

问题描述：
　　给定一个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

解题思路：
  可以用DFS解决，先放白色棋子，当放够n个白色棋子时，即白色棋子找到一种摆放方式；然后再放黑色棋子。放棋子时要做标记，用2表示白色棋子，3表示黑色棋子。当黑色棋子在白色棋子摆放好的情况下也找到一种摆放方式时，进行统计。
  判断是否在一对角线时，可以通过一个规律判断：两个点的行减列（x-y）相同就说明在同一主对角线；两个点的行加列（x+y）相同就说明在同一副对角线。

  这题是根据n皇后问题演变出来的，先看n皇后问题更容易理解这题。

代码如下：

import java.util.*;

public class 蓝桥杯2n皇后问题 {
	public static void main(String[] args){
		Scanner scanner = new Scanner(System.in);
		n = scanner.nextInt();
		int[][] arr = new int[n][n];
		int[] white = new int[n]; 
		int[] black = new int[n];
		for(int i=0; i<n; i++){
			white[i] = Integer.MAX_VALUE;
			black[i] = Integer.MAX_VALUE;
		}
		for(int i=0; i<n; i++){
			for(int j=0; j<n; j++){
				arr[i][j] = scanner.nextInt();
			}
		}
		dfs(arr, white, black, 0);
		System.out.println(count);
	}
	static int n;
	static int count=0;  //统计
	static boolean flag = true; //用于标记是在放白棋子还是黑棋子
	private static void dfs(int[][] arr, int[] white, int[] black, int row) {
		if(row==white.length){
			if(flag){
				flag = !flag;
				//用交换参数位置，进行黑色棋子的摆放
				dfs(arr, black, white, 0);//白色棋子放完后，开始放黑色棋子
				flag = !flag;//回溯
			}else{//两种棋子都放完了
				count++;			
			}
			return ;
		}
		for(int i=0; i<n; i++){
			if(check(arr,white, row, i)){
				white[row] = i;
				if(flag)
					arr[row][i] = 2;
				else
					arr[row][i] = 3;
				dfs(arr, white, black, row+1);
				white[row] = Integer.MAX_VALUE;
				arr[row][i] = 1;//回溯
			}
		}
	}
	
	private static boolean check(int[][] arr, int[] a,int x, int y) {
		if(arr[x][y]==2)return false;//白色棋子占了
		if(arr[x][y]==0) return false;//这个位置不能放皇后
		for(int i=0; i<a.length; i++){
			if(a[i] == y)//检查列
				return false;
			if(i-a[i] == x-y)//检查主对角线
				return false;
			if(i+a[i] == x+y)//检查副对角线
				return false;	
		}
		return true;
	}
}

提交结果满分：