leetcode-51-N皇后

原题: N皇后

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

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

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

示例 1：

输入：n = 4
输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

示例 2：

输入：n = 1
输出：[["Q"]]

提示：

• 1 <= n <= 9
• 皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

思路：

什么是n皇后？
在n×n格的棋盘上放置彼此不受攻击的n个皇后。按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。n皇后问题等价于再n×n的棋盘上放置n个皇后，任何2个皇后不妨在同一行或同一列或同一斜线上。

剪枝 + 回溯 解决
思路：
1、查找第一行，从左往右查找，发现合适的位置，放下第一个皇后
2、查找第二行，从左往右查找，发现合适的位置，放下第二个皇后
3、查找第三行，从左往右查找，发现合适的位置，放下第三个皇后
。
。
。
n、查找第n行，从左往右查找，发现合适的位置，放下第n个皇后

如何判断合适的位置？
假设要在m行k列放置一个皇后，只需要查看第一行到m-1行就行。如果k列已经存在皇后，那么m行k列这个位置就不适合了，如果这个位置的对角线上有其他皇后了，同样这个位置也不适合了。

方法一：递归 + 回溯+ 剪枝

public List<List<String>> solveNQueens(int n) {
    queen(n);
    return res;
}

List<List<String>> res = new ArrayList<>();
//数组的索引为行号，值为列号
int[] cols;
//n皇后的摆法总数
int ways;

public void queen(int n) {
    if (n < 1) {
        return;
    }
    cols = new int[n];
    place(0);
}
//在第row行放置皇后
private void place(int row) {
    //如果全部放完，则记录下一种摆法
    if (row == cols.length) {
        ways++;
        res.add(generateAns());
        return;
    }
     //遍历当前行的每一列
    for (int i = 0; i < cols.length; i++) {
        //如果第row行第i列可以放皇后
        if (isValid(row, i)) {
            cols[row] = i;
            //递归在row+1行放置皇后
            place(row + 1);
        }
    }

}
//判断在row行col列是否可以放置皇后
private boolean isValid(int row, int col) {
    for (int i = 0; i < row; i++) {
        //如果col列已经被row前面的行放置过皇后了，说明这个位置不能放置了
        if (cols[i] == col) {
            return false;
        }
        //row行col列这个位置的对角线存在别的皇后了，说明这个位置不能放置了
        //利用斜率 两个点的行的差值 等于 列的差值， 对角线斜率为 |k| = 1
        if (row - i == Math.abs(col - cols[i])) {
            return false;
        }
    }
    return true;
}
//记录每一种摆法
public List<String> generateAns() {
    List<String> res = new ArrayList<>();
    for (int row = 0; row < cols.length; row++) {
        StringBuilder sb = new StringBuilder();
        for (int col = 0; col < cols.length; col++) {
            if (cols[row] == col) {
                sb.append("Q");
            } else {
                sb.append(".");
            }
        }
        res.add(sb.toString());
    }
    return res;
}

方法二：位运算(后续更新…)