OJ练习第83题——N 皇后 II

N 皇后 II

力扣链接：52. N 皇后 II

题目描述

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

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

示例

思路

皇后的走法是：可以横直斜走，格数不限。因此要求皇后彼此之间不能相互攻击，等价于要求任何两个皇后都不能在同一行、同一列以及同一条斜线上。
暴力枚举复杂度非常高，因此可用回溯。

Java代码

class Solution {
    int count = 0;
    public int totalNQueens(int n) {
        int[][] res = new int[n][n];
        for(int row = 0; row < n; row++) {
            for(int col = 0; col < n; col++) {
                res[row][col] = 0;
            }
        }
        dfs(res, n, 0, 0);
        return count;
    }
    public void dfs(int[][] res, int n, int row, int sum) {
        if(sum == n) {
            count++;
            return;
        }
        for(int col = 0; col < n; col++) {
            if(iss(res, col, row, n) == false) continue;
            res[row][col] = 1;
            dfs(res, n, row + 1, sum + 1);
            res[row][col] = 0;
        }
    }
    public boolean iss(int[][] res, int col, int row, int n) {
        for(int i = 0; i < row; i++) {
            if(res[i][col] == 1) return false;
        }
        for(int i = 0; i < col; i++) {
            if(res[row][i] == 1) return false;
        }
        for(int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
            if(res[i][j] == 1) return false;
        }
        for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
            if(res[i][j] == 1) return false;
        }
        return true;
    }
}

---

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