51. N 皇后

Java实现

class Solution {
    /**
     * 数组索引是行号，数组元素是列号
     */
    int[] cols;   //存储每一行的皇后的位置（即列号）
    /**
     * 一共有多少种摆法
     */
    int ways;

    List<List<String>> res = new ArrayList<>();
    public List<List<String>> solveNQueens(int n) {
        cols = new int[n];
        place(0);
        return res;
    }

    /**
     * 从第row行开始摆放皇后
     * @param row
     */
    void place(int row) {
        if (row == cols.length) {  //如果行号等于数组长度，则表示行号越界，即摆放完毕。
            ways++;
            show();
            return;
        }

        for (int col = 0; col < cols.length; col++) {
            if (isValid(row, col)) {     //剪枝
                // 如果当前位置合法，则在第row行第col列摆放皇后
                cols[row] = col;
                place(row + 1); //从下一行开始摆放
                //上面那一句如果执行完，就回溯。
            }
        }
    }

    /**
     * 判断第row行第col列是否可以摆放皇后
     */
    boolean isValid(int row, int col) {  //col表示列数
        for (int i = 0; i < row; i++) {	    //从第0行开始，到当前行的前一行，
            // 第col列已经有皇后
            if (cols[i] == col) {   //如果第i行存储的皇后列数与所给列数一致，则表示该位置不可以摆放
                return false;
            }
            // 第i行的皇后跟第row行第col列格子处在同一斜线上，即斜率为1
            if (row - i == Math.abs(col - cols[i])) {    //cols[i]表示第i行存储的皇后在哪一列
                return false;
            }
        }
        return true;
    }
    void show() {
        List<String> tmp = null;
        tmp = new ArrayList<>();
        for (int row = 0; row < cols.length; row++) {
            StringBuilder temp = new StringBuilder();

            for (int col = 0; col < cols.length; col++) {
                String s = "";

                if (cols[row] == col) {
                    s = "Q";
                } else {
                    s = ".";
                }
                temp.append(s);

            }
//            System.out.println(temp);
//            System.out.println();
            tmp.add(temp.toString());

        }
        res.add(tmp);

    }
}