【蓝桥杯】 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

分析：每行放置黑白两个皇后，他们互不影响。在判断皇后时，黑的和黑的判断，白的和白的判断，因此，可以考虑先将白的皇后放置，然后再放置黑的，如果满足条件，则计数器加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);
}

}