经典算法题型(三):N皇后问题

    xiaoxiao2025-03-14  32

    一、基本概念

    1.N皇后问题是由国际象棋8皇后问题引申过来的,在国际象棋中有8行8列的棋盘,其中皇后的前进路线为:横、纵、以及双斜线,为了使皇后之间互不冲突,就要摆放皇后,使之前进路线不重叠。

    如下所示就是一个8皇后问题的解决方案,如图可以验证,各个皇后之间互不冲突:

    Q  .    .    .    .    .    .    . .    .    .    .   Q   .    .    . .    .    .    .    .    .    .   Q .    .    .    .    .   Q   .    . .    .   Q   .    .    .    .    . .    .    .    .    .    .   Q   . .    Q   .   .    .    .    .    .  .     .   .   Q   .    .    .    .

     2.由此引申到N皇后问题,借用leetcode 51的问题解决该问题

    二、问题解析:

    /*51. N皇后 n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。

    上图为 8 皇后问题的一种解法。

    给定一个整数 n,返回所有不同的 n 皇后问题的解决方案。

    每一种解法包含一个明确的 n 皇后问题的棋子放置方案,该方案中 'Q' 和 '.' 分别代表了皇后和空位。

    示例:

    输入: 4 输出: [  [".Q..",  // 解法 1   "...Q",   "Q...",   "..Q."],

     ["..Q.",  // 解法 2   "Q...",   "...Q",   ".Q.."] ] 解释: 4 皇后问题存在两个不同的解法。 * * */

    分析

    逐行摆放皇后的过程中的,如何判断某一个位置不能再摆放位置?

    1.使用一维数组,标记某一行、某一列是否被占用

    private boolean[] col;

    如果有皇后摆放在第一行或者第一列,则col[1]==true

    2.标记两个对角线,这里涉及到两个小技巧

    (1)右上:同一对角线相加为同一个数

    (2)左下:同一对角线相加为同一个数

    为了使其数组下标从0开始,需要在加上n-1;

     

     

    三、代码及注释

    import java.util.Arrays; import java.util.LinkedList; import java.util.List; import java.util.ArrayList; public class Solution { private boolean[] col; private boolean[] dia1; private boolean[] dia2; private ArrayList<List<String>> res; public List<List<String>> solveNQueens(int n) { res = new ArrayList<List<String>>(); col = new boolean[n]; dia1 = new boolean[2 * n - 1]; dia2 = new boolean[2 * n - 1]; LinkedList<Integer> row = new LinkedList<Integer>(); putQueen(n, 0, row); return res; } // 尝试在一个n皇后问题中, 摆放第index行的皇后位置 private void putQueen(int n, int index, LinkedList<Integer> row){ if(index == n){ res.add(generateBoard(n, row)); return; } for(int i = 0 ; i < n ; i ++) // 尝试将第index行的皇后摆放在第i列 if(!col[i] && !dia1[index + i] && !dia2[index - i + n - 1]){ row.addLast(i); col[i] = true; dia1[index + i] = true; dia2[index - i + n - 1] = true; putQueen(n, index + 1, row); col[i] = false; dia1[index + i] = false; dia2[index - i + n - 1] = false; row.removeLast(); } return; } private List<String> generateBoard(int n, LinkedList<Integer> row){ assert row.size() == n; ArrayList<String> board = new ArrayList<String>(); for(int i = 0 ; i < n ; i ++){ char[] charArray = new char[n]; Arrays.fill(charArray, '.'); charArray[row.get(i)] = 'Q'; board.add(new String(charArray)); } return board; } private static void printBoard(List<String> board){ for(String s: board) System.out.println(s); System.out.println(); } public static void main(String[] args) { int n = 4; List<List<String>> res = (new Solution()).solveNQueens(n); for(List<String> board: res) printBoard(board); } }

     

    最新回复(0)