Leetcode---N-Queens II

N皇后问题求解

最新推荐文章于 2019-07-17 21:57:48 发布

原创最新推荐文章于 2019-07-17 21:57:48 发布 · 510 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

Leetcode 专栏收录该内容

59 篇文章

订阅专栏

本文介绍了一种解决N皇后问题的方法，通过递归深度优先搜索（DFS）算法来计算特定大小棋盘上非攻击性皇后配置的数量。文章详细展示了如何通过检查每一步是否会导致皇后互相攻击来剪枝无效路径。

N皇后问题，求数量，比记录轨迹简单。

int N;
int sum;
vector<vector<int> > m;
bool check(int row,int column){
            if(row==1) return true;
            int i,j;
            for(i=0;i<=row-2;i++){
                if(m[i][column-1]==1) return false;
            }
            i = row-2;
            j = i-(row-column);
            while(i>=0&&j>=0){
                if(m[i][j]==1) return false;
                i--;
                j--;
            }
            i = row-2;
            j = row+column-i-2;
            while(i>=0&&j<=N-1){
                if(m[i][j]==1) return false;
                i--;
                j++;
            }
            return true;
        }
void dfs(int level){
    if(level==N){
        sum++;
    }
    else{
        for(int i=0;i<N;i++){
            if(check(level+1,i+1)){
                m[level][i]=1;
                dfs(level+1);
                m[level][i]=0;
            }
        }
    }
}
int totalNQueens(int n) {
    N=n;
    sum=0;
    for(int i=0;i<n;i++){
            vector<int> a(n,0);
            m.push_back(a);
        }
    dfs(0);
    return sum;
}