蓝桥杯——2n皇后问题 递归

n皇后问题

Description

在N×N的棋盘上放置N个皇后，使他们不相互攻击（即任意两个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框呈45度角的斜线上），对于给定的N，求出有多少种合理的放置方法。

Input

输入一个正整数N，N<=10，表示棋盘和皇后的数量。
8

Output

输出一个正整数，表示皇后的不同放置方法数量。
92

import java.util.Scanner;

public class Main {
	public static int n;
	static int queenPos[] = new int [100];    //queenPos[i] = j; i表示行数，j表示列
	//用来存放算好的皇后为止，最上角是（0，0）
	static void NQueen(int k) {  //在0~k-1行皇后已经摆好的情况下，摆放第k行以及之后的皇后
		int i;
		if(k == n) {  //n个皇后都已经摆好
			for(i = 0; i < n; i++)
				System.out.print((queenPos[i]+1) + " ");  //将已经摆好的皇后输出
				System.out.println();
				return;
		}
		
		for(i = 0; i < n; i++) {    //一一遍历第k个皇后的位置  代表列数
			int j;
			for(j = 0; j < k; j++) {    //代表行数, 和已经摆好的皇后的位置进行比较，看是否冲突
				if(queenPos[j] == i || Math.abs(queenPos[j]-i) == Math.abs(k-j)) {
					break;    //冲突，试下一个位置
				}
			}
			if(j == k) {  //如果不冲突的话
				queenPos[k] = i;  //将第k个皇后摆放在位置i
				NQueen(k+1);
			}
		}
	}
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		NQueen(0);
	}
}

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

---

在n皇后的基础上进行修改，先计算白皇后，在白皇后排列好的情况下，再排黑皇后，保证为0处和白皇后摆放处不能再摆放黑皇后。

import java.util.Scanner;

public class Main {
	public static int n;
	public static int count = 0;
	public static int[][] c = new int[100][100];
	static int wqueenPos[] = new int [100];    //queenPos[i] = j; i表示行数，j表示列
	static int bqueenPos[] = new int [100];
	//用来存放算好的皇后为止，最上角是（0，0）
	static void BlackQueen(int k) {
		int i,j;
		if(k == n) {
			count ++;
			return;
		}
		for(i = 0; i < n; i++) {    //一一遍历第k个皇后的位置  代表列数
			if(c[k][i] == 0) continue;   //棋盘为0处不能放置皇后
			if(wqueenPos[k] == i) continue;  //判断是不是白皇后摆放的地方，如果是 则跳过
			for(j = 0; j < k; j++) {    //代表行数, 和已经摆好的皇后的位置进行比较，看是否冲突   
				if(bqueenPos[j] == i || Math.abs(bqueenPos[j]-i) == Math.abs(k-j)) {
					break;    //冲突，试下一个位置
				}
			}
			if(j == k) {  //如果不冲突的话
				bqueenPos[k] = i;  //将第k个皇后摆放在位置i
				BlackQueen(k+1);
			}
		}
	}
	
	static void WhiteQueen(int k) {  //在0~k-1行皇后已经摆好的情况下，摆放第k行以及之后的皇后
		int i,j;
		if(k == n) {  //n个皇后都已经摆好
			BlackQueen(0);	
			return;
		}
		
		for(i = 0; i < n; i++) {    //一一遍历第k个皇后的位置  代表列数
			if(c[k][i] == 0) continue;   //棋盘为0处不能放置皇后
			for(j = 0; j < k; j++) {    //代表行数, 和已经摆好的皇后的位置进行比较，看是否冲突
				if(wqueenPos[j] == i || Math.abs(wqueenPos[j]-i) == Math.abs(k-j)) {
					break;    //冲突，试下一个位置
				}
			}
			if(j == k) {  //如果不冲突的话
				wqueenPos[k] = i;  //将第k个皇后摆放在位置i
				WhiteQueen(k+1);
			}
		}
	}
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j ++) {
				c[i][j] = sc.nextInt();
			}
		}
		WhiteQueen(0);
		System.out.println(count);
	}
}