蓝桥杯基础练习VIP-2n皇后问题 （Java代码）

解题思路:

注意事项:

参考代码:

import java.util.Scanner;

public class Main{

private static int[][] a;

private static int m;

private static int[][] v=new int[10][10];

private static int count;

public static void main(String[] args) {

Scanner input=new Scanner(System.in);

m=input.nextInt();

a=new int[m+1][m+1];

for (int i = 0; i <m; i++) {

for (int j = 0; j <m; j++) {

a[i][j]=input.nextInt();

}

}

dfs(0,2);

System.out.println(count);

}

private static void dfs(int x, int mun) {

if(x==m){

if(mun==2){

dfs(0,3);

}

if(mun==3){

count++;

}

}else{

for (int i = 0; i <m; i++) {

if(a[x][i]==1&&check(x,i,mun)&&v[x][i]==0){

v[x][i]=1;

a[x][i]=mun;

dfs(x+1,mun);

a[x][i]=1;

v[x][i]=0;

}

}

}

}

private static boolean check(int x,int y, int mun) {

for (int i = 0; i <x; i++) {

if(a[i][y]==mun)return false;

}

for (int i = x-1,j = y-1; i>=0&&j>=0; i--,j--) {

if(a[i][j]==mun)return false;

}

for (int i = x-1,j = y+1; i>=0&&j<m; i--,j++) {

if(a[i][j]==mun)return false;

}

return true;

}

}