Java解决八皇后、N皇后问题(包含例题、视频讲解链接)

八皇、N皇后后问题

八皇后问题介绍:

八皇后问题,是一个古老而著名的问题,是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯·贝瑟尔于 1848 年提出:在 8×8 格的国际象棋上摆放八个皇后,使其不能互相攻击,即:任意两个皇后都不能处于同一行、 同一列或同一斜线上,问有多少种摆法(92)。

算法思路分析:

在这里插入图片描述

  • 第一个皇后先放第一行第一列
  • 第二个皇后放在第二行第一列、然后判断是否 OK, 如果不 OK,继续放在第二列、第三列、依次把所有列都 放完,找到一个合适
  • 继续第三个皇后,还是第一列、第二列……直到第 8 个皇后也能放在一个不冲突的位置,算是找到了一个正确 解
  • 当得到一个正确解时,在栈回退到上一个栈时,就会开始回溯,即将第一个皇后,放到第一列的所有正确解, 全部得到
  • 然后回头继续第一个皇后放第二列,后面继续循环执行 1,2,3,4 的步骤
  • 示意图:

在这里插入图片描述

  • 说明:

理论上应该创建一个二维数组来表示棋盘,但是实际上可以通过算法,用一个一维数组即可解决问题。arr[8] = {0 , 4, 7, 5, 2, 6, 1, 3} //对应 arr 下标 表示第几行,即第几个皇后,arr[i] = val, val 表示第 i+1 个皇后,放在第 i+1 行的第 val+1 列。

韩老师的代码

public class Queue8 {
    
    
	public static void main(String[] args) {
    
    
		// 测试一把 , 8皇后是否正确
		Queue8 queue8 = new Queue8();
		queue8.check(0);
		System.out.printf("一共有%d解法", count);
		System.out.println();
		System.out.printf("一共判断冲突的次数%d次", judgeCount); // 1.5w
	}
	
	/**
	 * 说明:理论上应该创建一个二维数组来表示棋盘,但是实际上可以通过算法,用一个一维数组即可解决问题. 
	 * arr[8] = {0 , 4, 7, 5, 2, 6, 1, 3}
	 * 对应arr 下标 表示第几行,即第几个皇后, arr[i] = val , val 表示第i+1个皇后,放在第i+1行的第val+1列
	 */
	// 定义一个max表示共有多少个皇后
	int max = 8;// 8个皇后摆放在8*8棋盘
	// 定义数组array, 保存皇后放置位置的结果,比如 arr = {0 , 4, 7, 5, 2, 6, 1, 3}
	int[] array = new int[max];
	static int count = 0;// 记录共有多少中摆放方式
	static int judgeCount = 0;// 记录判断了多少次

	/**
	 * 编写一个方法,放置第n个皇后 特别注意: check 是 每一次递归时, 进入到check中都有 for(int i = 0; i < max;i++)
	 * 因此会有回溯
	 * 
	 * @param n
	 */
	private void check(int n) {
    
    
		if (n == max) {
    
    // n = 8: 第n+1=9个皇后, 说明前8个皇后就已经放好
			print(); // 输出该情况的摆法
			return;// 结束
		}

		// 依次放入皇后,并判断是否冲突
		for (int i = 0; i < max; i++) {
    
    
			// 先把当前这个皇后 n , 放到该行的第(i+1)列
			array[n] = i;
			// 判断当放置第n个皇后到i列时,是否冲突
			if (judge(n)) {
    
     // 不冲突
				// 若不冲突,就接着放n+1个皇后,即开始递归
				check(n + 1); //
			}
			// 如果冲突,就继续执行 array[n] = i; 即将第n个皇后,放置在本行得 后移的一个位置
		}
	}

	/**
	 * 判断当我们放置第n个皇后, 就去检测该皇后是否和前面已经摆放的皇后冲突
	 * 
	 * @param n 表示第n个皇后
	 * @return
	 */
	private boolean judge(int n) {
    
    
		judgeCount++;
		for (int i = 0; i < n; i++) {
    
    
			// 说明
			// 1. array[i] == array[n] 表示判断 第n个皇后是否和前面的n-1个皇后在同一列
			// 2. Math.abs(n-i) == Math.abs(array[n] - array[i]) 表示判断第n个皇后是否和第i皇后是否在同一斜线
			// n = 1 放置第 2列:1 <---> n = 1 array[1] = 1
			// Math.abs(1-0) == 1 同理 Math.abs(array[n] - array[i]) = Math.abs(1-0) = 1
			// 3. 判断是否在同一行, 没有必要,因为 n 每次都在递增
			if (array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])) {
    
    
				return false;
			}
		}
		return true;
	}

	// 写一个方法,可以将皇后摆放的位置输出
	private void print() {
    
    
		count++;
		for (int i = 0; i < array.length; i++) {
    
    
			System.out.print(array[i] + " ");
		}
		System.out.println();
	}
}

力扣皇后相关题目

public class Test36 {
    
    
	public static void main(String[] args) {
    
    
		// 测试max=4
		System.out.println(new Test36().solveNQueens(4));
	}

	public List<List<String>> solveNQueens(int max) {
    
    
		/**
		 * 说明:理论上应该创建一个二维数组来表示棋盘,但是实际上可以通过算法,用一个一维数组即可解决问题. 
		 * arr[8] = {0 , 4, 7, 5, 2, 6, 1, 3}
		 * 对应arr 下标 表示第几行,即第几个皇后, arr[i] = val , val 表示第i+1个皇后,放在第i+1行的第val+1列
		 */
		// 定义一个max表示共有多少个皇后
		int[] array = new int[max];// max个皇后摆放在max*max棋盘
		
		// 返回的结果集
		// eg:4个皇后时输出 -> [[.Q.., ...Q, Q..., ..Q.], [..Q., Q..., ...Q, .Q..]]
		List<List<String>> res = new ArrayList<>();
		// 放置第0(即第1个皇后),最多放置max个皇后,一维数组array表示的二维棋盘,需要返回的结果集
		check(0, max, array, res);
		return res;
	}

	/**
	 * 回溯法:放置第n个皇后
	 * @param n 放置第n+1个皇后(n从0开始)
	 * @param max 最多放置max个皇后
	 * @param array 一维数组array表示的二维棋盘
	 * @param res 需要返回的结果集
	 */
	private void check(int n, int max, int[] array, List<List<String>> res) {
    
    
		if (n == max) {
    
     // n+1个皇后已经摆放完成,说明前n个皇后就已经放好了
			List<String> list = print(array);// 将该情况的摆法存入List<String>
			res.add(list);// 将该情况的摆法存入结果集List<List<String>>
			return;// 结束
		}

		// 依次放入皇后,并判断是否冲突
		for (int i = 0; i < max; i++) {
    
    
			// 先把当前这个皇后 n , 放到该行的第(i+1)列
			array[n] = i;// 第一个皇后在第一行第一列所有情况出现后,依次放置第n列
			if (judge(n, array)) {
    
    // 判断当放置第n个皇后到i列时,是否冲突
				// 若不冲突,就接着放n+1个皇后,即开始递归
				check(n + 1, max, array, res);
			}
			// 如果冲突,就继续执行 array[n] = i; 即将第n个皇后,放置在本行得 后移的一个位置
		}
	}

	/**
	 * 判断摆放第n个皇时,和之前的皇后是否冲突
	 * @param n
	 * @param array
	 * @return
	 */
	private boolean judge(int n, int[] array) {
    
    
		for (int i = 0; i < n; i++) {
    
    
			// 说明
			// 1. array[i] == array[n] 表示判断 第n个皇后是否和前面的n-1个皇后在同一列
			// 2. Math.abs(n-i) == Math.abs(array[n] - array[i]) 表示判断第n个皇后是否和第i皇后是否在同一斜线
			// n = 1 放置第 2列:1 <---> n = 1 array[1] = 1
			// Math.abs(1-0) == 1 同理 Math.abs(array[n] - array[i]) = Math.abs(1-0) = 1
			// 3. 判断是否在同一行, 没有必要,因为 n 每次都在递增
			if (array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])) {
    
    
				return false;
			}
		}
		return true;
	}

	/**
	 * 将某种情况下的摆放方式存入List<String>并返回
	 * @param array
	 * @return
	 */
	private List<String> print(int[] array) {
    
    
		List<String> list = new ArrayList<String>();
		for (int i = 0; i < array.length; i++) {
    
    // array = [1 3 0 2]
			String str = "";
			for (int j = 0; j < array.length; j++) {
    
    
				if (j == array[i]) {
    
    
					str = str + "Q";
				} else {
    
    
					str = str + ".";
				}
			}
			list.add(str);
		}
		return list;
	}
}

猜你喜欢

转载自blog.csdn.net/weixin_43591980/article/details/115344903