LeetCode_52_N皇后Ⅱ

题目链接

• https://leetcode.cn/problems/n-queens-ii/

题目描述

按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

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

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

示例 1：

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

示例 2：

输入：n = 1
输出：1

提示：

• 1 <= n <= 9

解题思路

直接使用N皇后的代码，然后返回数量

AC代码

class Solution {
    
    public int totalNQueens(int n) {
    
        return solveNQueens(n).size();
    }

    public static List<List<String>> solveNQueens(int n) {
    
        List<List<String>> ans = new ArrayList<>();
        char[][] chess = new char[n][n];
        for (char[] c : chess) {
    
            Arrays.fill(c, '.');
        }
        backTracing(n, 0, chess, ans);
        return ans;
    }

    private static void backTracing(int n, int row, char[][] chess, List<List<String>> ans) {
    
        if (row == n) {
    
            ans.add(Array2List(chess));
            return;
        }
        for (int col = 0; col < n; col++) {
    
            if (isValid(row, col, n, chess)) {
    
                chess[row][col] = 'Q';
                backTracing(n, row + 1, chess, ans);
                chess[row][col] = '.';
            }
        }
    }

    private static List<String> Array2List(char[][] chess) {
    
        List<String> list = new ArrayList<>();
        for (char[] c : chess) {
    
            list.add(String.copyValueOf(c));
        }
        return list;
    }

    private static boolean isValid(int row, int col, int n, char[][] chess) {
    
        //检查列
        for (int i = 0; i < row; i++) {
    
            if (chess[i][col] == 'Q') {
    
                return false;
            }
        }
        //检查45度对角线
        for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
    
            if (chess[i][j] == 'Q') {
    
                return false;
            }
        }
        //检查135度对角线
        for (int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
    
            if (chess[i][j] == 'Q') {
    
                return false;
            }
        }
        return true;
    }
}