1581:基础练习 2n皇后问题

时间:2023-02-12 22:42:33

//做题(copy别人代码)二十分钟,debug两小时:)mmp...

明后天再看一遍吧,实在现在恶心看不进去了orz

Description

给定一个n*n的棋盘,棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后
和n个白皇后,使任意的两个黑皇后都不在同一行、同一列或同一条对角线上,任意的两
个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法?n小于等于8。

Input

输入的第一行为一个整数n,表示棋盘的大小。
接下来n行,每行n个0或1的整数,如果一个整数为1,表示对应的位置可以放皇后,

如果一个整数为0,表示对应的位置不可以放皇后。

Output

  输出一个整数,表示总共有多少种放法。

Sample Input

No.1
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1

No.2
4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1

Sample Output

No.1
2

No.2
0
import java.util.Scanner;
public class Main {  
	public static int cnt=0,black=2,white=3;//0表示不可放皇后,1可放,2黑,3白
	public static int n;
	public static int [][]queen=null;
	public static boolean judgeQueen(int i,int j,int which){
		//which就是指查找的queen的颜色
		int r,c;
		for(r=0;r<n;r++)//上下
			if(queen[r][j]==which)
				return false;
		for(c=0;c<n;c++)//左右
			if(queen[i][c]==which)
				return false;
		for(r=i-1,c=j-1;r>=0&&c>=0;r--,c--)//左上角
			if(queen[r][c]==which)
				return false;
        for(r=i+1,c=j+1;r<n&&c<n;r++,c++)     //右下角
            if(queen[r][c]==which)
                return false;
        for(r=i-1,c=j+1;r>=0&&c<n;r--,c++)     //右上角
            if(queen[r][c]==which)
                return false;
        for(r=i+1,c=j-1;r<n&&c>=0;r++,c--)     //左下角
            if(queen[r][c]==which)
                return false;
        
        return true;
	}
	static void set(int i,int which){//先黑后白
		if(i>=n){//全部找完
			if(which==0){//找完一种情况
				//这里错了,看的代码是0
				cnt++;
				which=1;
				return;
			}
			which=0;i=0;//改了这里
		}
		if(which!=0){//这里改了
			for(int j=0;j<n;j++)
				if(judgeQueen(i,j,black)&&queen[i][j]==1){
					//我把这个which改成black就运行正常了!我的妈!
					queen[i][j]=black;
					//System.out.println("queen"+i+" "+j+"=black");
					set(i+1,which);
					queen[i][j]=1;
					//这里为啥子是1呢,真心不懂,这时候应该i已经超过了才对,如果设成1会不会把不能放的位置给设成可放置了呢
				}
		}
		else{
			for(int j=0;j<n;j++){
				if(judgeQueen(i,j,white)&&queen[i][j]==1){
					//我把这个which改成black就运行正常了!我的妈!
					queen[i][j]=white;//我踏马这里没改!!!
					set(i+1,which);
					queen[i][j]=1;
				}
			}			
		}
	}
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		n=sc.nextInt();
		queen=new int[n][n];
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
				queen[i][j]=sc.nextInt();
		cnt=0;
		set(0,black);
		System.out.println(cnt);
	}
}