1. 程式人生 > >回溯法解決2n皇后(8皇后)問題

回溯法解決2n皇后(8皇后)問題

8皇后問題是演算法入門的經典,在8*8的國際象棋上擺放八個皇后,使其不能相互攻擊,即任意兩個皇后都不能處於同一

今天要說的2n皇后問題與傳統的8皇后問題有些不同,在一個n*n的棋盤中,有些位置不允許放皇后,現在要向棋盤中

放入n個黑皇后和n個白皇后,使任意的兩個黑白皇后都不在同一行,同一列或者同一條對角線上。看起來問題變得有

些複雜了,但其實本質上還是8皇后問題。

我們使用回溯演算法和遞迴的思想來解決這個問題。每行只能放置一個黑皇后和一個白皇后,從第一行第一列開始放

置,假設第一行第一列放置白皇后,那麼黑皇后在第一排的位置就還有n-1中可能,依次檢測每一種可能性,首先將

黑皇后放在第一排第二列的位置。然後放置第二排,在放第二排時就需要考慮到皇后的擺放規則,打個比方說在第二

排第一列放一個白皇后,那麼此時就需要進行檢測,白皇后的位置是否安全可行,明顯的一列只能放一個白皇后,所

以這個皇后的擺放位置不可行,重新擺放,在重新擺放之前會清除本行原本的擺放記錄。當這一排的所有位置都不能

安全的擺放兩個皇后時,就需要退回到上一行的擺放,清楚上一行原本的擺放記錄,重新擺放。

直到第一次擺放到最後一行,並且棋盤中的黑白皇后數量正確,此時才算是得出了一個正解。到這裡程式並不會結

束,因為得到一個正解並不等於嘗試完所有的可能性,程式只有在嘗試完所有的可能擺放皇后位置以後,才會結束。

瞭解演算法的思路以後程式碼的框架也就出來了,用一個二維陣列模擬棋盤,預設值為0和1,0即為此位置不可以擺放皇

後,在擺放皇后時也需要對此進行驗證,這裡我們用6代表白皇后,9代表黑皇后,放置皇后也就是改變陣列中指定位

置的數值。在驗證擺放皇后位置是否可行時,需要判斷三個方向是否有該種皇后,即兩條對角線和本列,而且只需要

考慮皇后所在行的上面,因為下面還沒有擺放皇后啊!為了便於處理,將棋盤分為三部分(除去第一行,第一行不需

要判斷),第一部分為第一列,第二部分為最後一列,第三部分為其他列。這樣分第一部分和第二部分只需要考慮一

條對角線,只有第三部分需要考慮兩條。

import java.util.Scanner;

public class Demo4 {

	private static int result = 0;
	

	public static boolean isSafety(short[][] chess, int row, int x, int queen) {
		
		if (row != 0 && x != 0 && x != chess.length - 1) {
			
			int step=1,step2=1;
			while(row-step>=0&&x+step<=chess.length-1){
				if(chess[row-step][x+step]==queen){
					return false;	
				}
				step++;
			}
			while(row-step2>=0&&x-step2>=0){
				if(chess[row-step2][x-step2]==queen){
					return false;	
				}
				step2++;
			}
		}
	
		if (row != 0 && x == 0) {
			int step=1;
			while(row-step>=0&&x+step<=chess.length-1){
				if(chess[row-step][x+step]==queen){
					return false;	
				}
				step++;
			}
		}

		if (row != 0 && x == chess.length-1) {
			int step=1;
			while(row-step>=0&&x-step>=0){
				if(chess[row-step][x-step]==queen){
					return false;	
				}
				step++;
			}
		}

		for (int i = 0; i < row; i++) {
			if (chess[i][x] == queen) {
				return false;	
			}
		}

		return true;
	}

	public static void putQueen(short[][] chess, int row) {

		// 擺完一盤
		if (row == chess.length) {
			if(queenNumber(chess)){
				result++;
			}	
			return;
		}

		for (int i = 0; i < chess.length; i++) {
			//清除原本的擺放記錄
			for (int j = 0; j < chess.length; j++) {
				if(chess[row][j]!=0){
					chess[row][j] = 1;
				}
			}

			if (isSafety(chess, row, i, 6)&&chess[row][i]==1) {
				//擺放白皇后
				chess[row][i] = 6;

				for (int k = 0; k < chess.length; k++) {
					//清除原本的擺放記錄
					for (int l = 0; l < chess.length; l++) {
						if (chess[row][l] != 0 && l != i) {
							chess[row][l] = 1;
						}
					}

					if (isSafety(chess, row, k, 9)&&chess[row][k]==1&&k != i) {
						//擺放黑皇后
						chess[row][k]=9;
						//擺放下一行
						putQueen(chess, row + 1);

					}
				}
			}
		}

	}

	public static boolean queenNumber(short[][] chess){
		int white=0,black=0;
		for(int i=0;i<chess.length;i++){
			for(int j=0;j<chess.length;j++){
				if(chess[i][j]==6){
					white++;
				}
				if(chess[i][j]==9){
					black++;
				}
				
			}
		}
		if(white==chess.length&&black==chess.length){
			return true;
		}
		return false;
	}

	public static void main(String[] args) {
		Scanner console = new Scanner(System.in);
		int n = console.nextInt();
		short[][] chess = new short[n][n];

		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				chess[i][j] = (short) console.nextInt();
			}
		}

		putQueen(chess, 0);
		System.out.println(result);

	}
}




控制檯Sample:

4
1 0 1 1 
1 1 1 1
1 1 1 1
1 1 1 1
0