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

import java.util.*;

//2n皇后问题

public class Main {
	public static final int no=0; // 代表不能放棋子
	public static final int yes=1; //代表可以放棋子
	public static final int hei=2; //代表黑棋子
	public static final int bai=3; //代表白棋子
	public static int n; // n行n列
	public static int[][] qipan=null;//代表棋盘 n*n的二维数组
	public static int count=0;//已经放上的当前皇后数
	public static int num=0; //摆放的2n皇后的总解
	
	/**
	 * 
	 * @param qizi 参数棋子
	 * @return 根据当前的棋子返回相反的棋子 如果当前是黑 就返回白
	 */
	public static int get(int qizi){
	  int num=qizi==2? 3: 2;
	  return num;
	}
	
	/**
	 * 
	 * @param k 行
	 * @param m 列
	 * @param qizi 所要检查的棋子
	 * @return 如果检查存在冲突 返回false 否则返回true
	 */
	public static boolean jc(int k,int m,int qizi){
		if(qipan[k][m]==get(qizi)||qipan[k][m]==no){ //如果当前列已经放了不同棋子或者当前位置是no
			return false;
		}
		
        for(int i=0;i<n;i++){  //扫描当前列 是否有存在相同棋子     
            if(qipan[i][m]==qizi){       
                  return false;            
            }         
        }
        
        int i=0,j=0;  
        for(i=k,j=m;i<n&&j<n;i++,j++){    //右下角；
             if(qipan[i][j]==qizi){              
                 return false;           
             }        
        }        
        for(i=k,j=m;i>=0&&j>=0;i--,j--){   //左上角；
           if(qipan[i][j]==qizi){     
                 return false;      
           }      
        }       
        for(i=k,j=m;i>=0&&j<n;i--,j++){ // 右上角；
             if(qipan[i][j]==qizi){  
                return false;    
             }       
        }         
        for(i=k,j=m;i<n&&j>=0;i++,j--){    //左下角；
             if(qipan[i][j]==qizi){ 
                 return false;  
             }       
        }       
		
		return true;
	}
	/**
	 * 扫描棋盘是否有n个黑和n个白棋子
	 */
	public static boolean sum(){
		int bainum=0;
		int heinum=0;
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				if(qipan[i][j]==bai)
					bainum++;
				if(qipan[i][j]==hei)
					heinum++;
			}
		}
		if(bainum==n&&heinum==n){
			num++;
			return true;
		}
		return false;
			
	}
	/**
	 * 递归方法的实现
	 * @param line 行号
	 * @param qizi 当前放的棋子
	 */
	public static void mk(int line , int qizi){
		if(line>=n) return ; //如果超出了界限n 就返回
		for(int i=0;i<n;i++){
			if(jc(line,i,qizi)){
				qipan[line][i]=qizi;
				count++;
				if(count>=n){
					if(sum()){
						print();
					}
					int c=count;
					count=0;
					mk(0,bai); //继续递归
					count=c;
				}
				mk(line+1,qizi);
				qipan[line][i]=yes; //回溯
				count--; 
			}
			
		}
	}
	
	/**
	 * 打印当前棋盘
	 */
	public static void print(){
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				System.out.print(qipan[i][j]+" ");
			}
			System.out.println();
		}
		System.out.println(":"+num);
	}
	
	public static void main(String args[]){
		Scanner scan=new Scanner(System.in);
		n=scan.nextInt();
		qipan=new int[n][n];
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				qipan[i][j]=scan.nextInt();
			}
		}
		mk(0,hei);
		
		System.out.println(num);
	}
}