Leetcode - N Queues II

原创于 2015-07-30 20:52:15 发布 · 91 阅读

0 ·

CC 4.0 BY-SA版权

Backtracking 专栏收录该内容

22 篇文章

订阅专栏

本文详细解析了如何使用深度优先搜索算法解决经典的N皇后问题，通过递归实现棋盘状态的枚举，确保每一行只有一个皇后，并检查对角线上是否已有皇后存在，最终计算出所有可能的解决方案数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[分析] 做完N皇后第一题，这个就so easy~


public class Solution {
    public int totalNQueens(int n) {
        int[] rows = new int[n]; // rows[i] = j, means board[i][j] = Q
        int[] cols = new int[n]; // cols[j] = i, means board[i][j] = Q
        for (int i = 0; i < n; i++) {
            rows[i] = -1;
            cols[i] = -1;
        }
        return recur(n, 0, rows, cols);
    }

    public int recur(int n, int r, int[] rows, int[] cols) {
        if (r == n) {
            return 1;
        }
        int result = 0;
        for (int j = 0; j < n; j++) {
            if (cols[j] == -1 && verifyDiagnal(rows, r, j)) {
                rows[r] = j;
                cols[j] = r;
                result += recur(n, r + 1, rows, cols);
                rows[r] = -1;
                cols[j] = -1;
            }
        }
        return result;
    }

    public boolean verifyDiagnal (int[] rows, int r, int c) {
        for (int i = 0; i < r; i++) {
            if (r - i == Math.abs(c - rows[i]))
                return false;
        }
        return true;
    }
}