Leetcode刷题之 【N皇后问题】

1 题目

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

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

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

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

示例：

输入：4
输出：[
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],

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

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-queens
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2 解题思路以及java代码

1 从棋盘从上向下遍历

使用一个一维数组表示棋盘的布局，由于棋盘的布局有规律，即正方形，且大小都是n，一维数组的下标表示行数，数组中的数字表示列数，如：第i行第nums[i]列。

• 每一行从第0列开始遍历，先将第col列放上皇后看看，然后判断第row行和第col列上放上皇后是否合法
• 合法的判别条件
  • 在同一行或同一列即col==row或nums[row]==nums[col]
  • 在对角线上，对于二维数组的对角线上的两个元素，其坐标如下规律：两个元素的行下标差的绝对值和列坐标差的绝对值相等，即abs(row-col)==abs(nums[row]-nums[col])

class Solution {
    private List<List<String>> res = new ArrayList<>();

    public List<List<String>> solveNQueens(int n){
        /**
         *nums表示棋盘，下标i表示行数，nuns[i]表示列数，从0开始计算
         */
        int []nums = new int[n];
        Arrays.fill(nums,-1);
        backTrack3(nums,0);
        return res;
    }

    /**
     *
     * @param nums: 期盼的行和列  nums的下标代表行，下标中存放的数据代表列，如：第i行第nums[i]列
     * @param row：当前的行数
     */
    private void backTrack3(int[]nums,int row){
        //1.退出条件
        if (row==nums.length){
            res.add(change(nums));
            return;
        }
        /**
         *
         */
        for (int col = 0; col < nums.length; col++) {//col代表列
            nums[row]=col;//表示第row行第col列已经放置了皇后，即第row行皇后所在的列
            /**
             * 将皇后放在第row行第col上看看，然后用isValid判断当前位置（row,col）上的皇后是否合法
             *  - 如果合法，则继续向下一行放置皇后
             *  - 如果不合法，则不放置，继续向下一列试探，以此类推
             */
            if (isValid(row,nums))//再检查第row是否可以放置，如果可以放置，则继续放置第row+1行
                backTrack3(nums,row+1);
        }

    }

    /**
     * 判断第row行是否可以放置皇后
     * @param row 行
     * @param nums 棋盘
     * @return
     */
    private boolean isValid(int row,int[]nums){
        for (int i = 0; i < row; i++) {//i表示row之前的行
            /**
             * 二维数组的对角线有如下规律：
             *        对角线上的元素的行下标之差的绝对值和列坐标之差的绝对值相等
             * 如果第row行已经放置了皇后，则不能再放了
             * nums[i]==nums[row]： 表示皇后在第i行所在的列和皇后在第row行所在的列相等，即他们会互相攻击，不合法
             * Math.abs(nums[row]-nums[i])==Math.abs(row-i)：表示第i行的皇后和第row行的皇后处在对角线上，也会互相攻击，不合法
             */
            if (nums[i]==nums[row] || Math.abs(nums[row]-nums[i])==Math.abs(row-i))
                return false;
        }
        return true;//表示皇后所处的位置合法
    }
    
    /**
    转化函数，将nums中存放的皇后位置转化为棋盘格式
    **/
    private List<String> change(int []nums){
        List<String> r = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            char[] chars = new char[nums.length];
            Arrays.fill(chars,'.');
            chars[nums[i]]='Q';
            r.add(new String(chars));
        }
        return r;
    }
}