LeetCode 101 N-Queens II

最新推荐文章于 2024-06-25 08:22:38 发布

JavyZheng

最新推荐文章于 2024-06-25 08:22:38 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： LeetCode N-Queens II 递归

本文链接：https://blog.youkuaiyun.com/javyzheng/article/details/40536163

LeetCode 专栏收录该内容

151 篇文章

订阅专栏

本文介绍了一种解决N皇后问题的方法，该方法通过深度优先搜索(DFS)算法计算在N×N的棋盘上放置N个皇后的所有不同方式的数量。文章提供了一个Java实现示例，展示了如何通过递归调用和有效性检查来完成这一任务。

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

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

分析：

这和上一题没有本质区别，只是把上一题的打印棋盘变成结果加1，所以准确来讲还简单一些，需要一个类变量来存储结果。

public class Solution {
    int res;
    public int totalNQueens(int n) {
        res = 0;
        if(n < 0) return res;
        int[] loc = new int[n];
        dfs(loc, 0, n);
        return res;
    }
    public void dfs(int[] loc, int cur, int n){
        if(cur == n){
            res++;
            return;
        }
        for(int i=0; i<n; i++){
            loc[cur] = i;
            if(isValid(loc, cur))
                dfs(loc, cur+1, n);
        }
    }
    
    public boolean isValid(int[] loc, int cur){
        for(int i=0; i<cur; i++){
            if(loc[i]==loc[cur] || Math.abs(loc[i]-loc[cur])==(cur-i))
                return false;
        }
        return true;
    }
}