Leetcode---N-Queens II

我把葡萄酿成酒

于 2015-04-11 17:18:42 发布

阅读量501

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： leetcode 算法

本文链接：https://blog.youkuaiyun.com/ffmpeg4976/article/details/44997057

Leetcode 专栏收录该内容

59 篇文章

订阅专栏

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

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

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;
}