算法基础——回溯和N皇后问题

算法基础——回溯和N皇后问题
问题描述：
力扣【51】
1.往一个n*n的棋盘里面放入n个皇后
2.要求一行只能一个皇后，一列只能一个皇后，两条对角线也只能放一个皇后
3.求皇后的放置有多少种
例如8皇后的放置方法92种
解题思路：
基于位运算的回溯
1.使用一个数组记录：
使用三个整数 columns、diagonals
1和diagonals 2分别记录每一列以及两个方向的每条斜线上是否有皇后，每个整数有 N个二进制位。棋盘的每一列对应每个整数的二进制表示中的一个数位，其中棋盘的最左列对应每个整数的最低二进制位，最右列对应每个整数的最高二进制位。
2.根据条件放置好皇后更新当前行皇后列的下标：

1. 初始时，三个整数的值都等于 0，表示没有放置任何皇后；
2. 在当前行放置皇后，如果皇后放置在第 i列，则将三个整数的第 i个二进制位（指从低到高的第 i 个二进制位）的值设为 1
3. 进入下一行时，columns 的值保持不变，diagonals1左移一位，diagonals2右移一位，由于棋盘的最左列对应每个整数的最低二进制位，即每个整数的最右二进制位，因此对整数的移位操作方向和对棋盘的移位操作方向相反（对棋盘的移位操作方向是diagonals1右移一位，diagonals 2左移一位）。

3.N个皇后放置完毕，找到一个可能的解

1. 每次放置皇后时，三个整数的按位或运算的结果即为不能放置皇后的位置，其余位置即为可以放置皇后的位置。然后遍历这些位置，尝试放置皇后并得到可能的解。
   4.当找到一个可能的解之后，将数组转换成表示棋盘状态的列表，并将该棋盘状态的列表加入返回列表。

class Solution {
    public List<List<String>> solveNQueens(int n) {
        int[] queens = new int[n];
        Arrays.fill(queens, -1);
        List<List<String>> solutions = new ArrayList<List<String>>();
        solve(solutions, queens, n, 0, 0, 0, 0);
        return solutions;
    }

    public void solve(List<List<String>> solutions, int[] queens, int n, int row, int columns, int diagonals1, int diagonals2) {
        if (row == n) {
            List<String> board = generateBoard(queens, n);
            solutions.add(board);
        } else {
            int availablePositions = ((1 << n) - 1) & (~(columns | diagonals1 | diagonals2));
            while (availablePositions != 0) {
                int position = availablePositions & (-availablePositions);
                availablePositions = availablePositions & (availablePositions - 1);
                int column = Integer.bitCount(position - 1);
                queens[row] = column;
                solve(solutions, queens, n, row + 1, columns | position, (diagonals1 | position) << 1, (diagonals2 | position) >> 1);
                queens[row] = -1;
            }
        }
    }

    public List<String> generateBoard(int[] queens, int n) {
        List<String> board = new ArrayList<String>();
        for (int i = 0; i < n; i++) {
            char[] row = new char[n];
            Arrays.fill(row, '.');
            row[queens[i]] = 'Q';
            board.add(new String(row));
        }
        return board;
    }
}