【Leetcode】N-Queens II

N皇后问题求解计数

最新推荐文章于 2025-09-09 20:07:09 发布

转载最新推荐文章于 2025-09-09 20:07:09 发布 · 153 阅读

文章标签：

#leetcode #java #arraylist #array

本文介绍了一种解决N皇后问题的方法，该方法通过深度优先搜索(DFS)算法来计算在N×N的棋盘上放置N个皇后的方式总数，确保任两个皇后不会在同一行、同一列或同一斜线上。

Follow up for N-Queens problem.

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

Java:

http://blog.youkuaiyun.com/u011095253/article/details/9158473

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+=1;
            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;
    }
}