算法练习普通(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.Scanner;

public class QueensProblem {
		static int a[],s[];
		static int arr[][];
		static int n;
		static int ac=0;
		static int q;
		
	public static void main(String[] args) {
		// 
		Scanner sc=new Scanner(System.in);
		n=sc.nextInt();
		a=new int[ n ];
		s=new int[ n ];
		arr=new int[ n + 1 ][ n + 1 ];
		
		for (int i = 1; i <= n; i++) 
			for (int j = 1; j <= n; j++) 
				arr[ i ][ j ]=sc.nextInt();
		
		sc.close();	
		
		// 先放黑皇后
		pa(0);
		System.out.println(q);
			
	}

	public static void pa(int m) {
		
		if( m == n ) {			
			// 放了一轮黑皇后，再放白皇后
			pas(0);
			return;
		}
		for (int i = 1; i <= n; i++) {
			// 如果是1就可以放皇后
			if(arr[ m + 1 ][ i ] != 0) {
				a[ m ] = i;		//放入一个皇后
				ac = 1;			// 1代表没有冲突
				
				for (int j = 0; j < m; j++) {
					// 同行同列是否有冲突
					if(a[ m ] == a[ j ])
						ac = 0;
					
					// 对角线上是否有冲突
					else if(Math.abs(a[ j ] - a[ m ]) == ( m - j ))
						ac = 0;
				}
				
				if( ac == 1) 
					pa( m + 1 );
			}
		}
	}
	

	public static void pas(int m) {
		if( m == n ) {
			q++;
			return;
		}
		
		for (int i = 1; i <= n; i++) {
			if(arr[ m + 1 ][ i ] == 1 && a[ m ] != i) {
				s[ m ] = i;
				ac = 1;	
				
			for (int j = 0; j < m; j++) {
				if(s[ m ] == s[ j ])
					ac = 0;
				else if(Math.abs(s[ j ] - s[ m ]) == ( m - j ))
						ac = 0;
			}
			
			if( ac == 1 ) 
				pas( m + 1 );
			
			}
			
		}
	}
	
}