130_N-Queens II

最新推荐文章于 2024-10-17 19:06:39 发布

ezzffzz

最新推荐文章于 2024-10-17 19:06:39 发布

阅读量326

点赞数

分类专栏： leetcode 算法编程文章标签： leetcode 算法编程语言面试递归

本文链接：https://blog.csdn.net/ezzffzz/article/details/38018731

版权

leetcode 同时被 3 个专栏收录

155 篇文章 0 订阅

订阅专栏

算法

155 篇文章 0 订阅

订阅专栏

编程

155 篇文章 0 订阅

订阅专栏

计算N皇后的个数

1:注意特殊情况；2:采用递归和回溯的方法；3:设置数组保持每一行皇后摆放的位置，并计算当前位置是否满足题意

    int totalNQueens(int n)
    {
        if( n <= 0 )
        {
            return 0;
        }
        
        vector<int> pos(n, 0);
        int totalNumber = 0;
        
        for(int i = 0; i < n; i++)
        {
            pos[0] = i;
            totalNQueensCore(n, 1, totalNumber, pos);
        }

        return totalNumber;
    }
    
    void totalNQueensCore(int n, int curIndex, int &totalNumber, vector<int> &pos)
    {
        if(curIndex == n)
        {
            totalNumber += 1;
            return;
        }
        
        for(int i = 0; i < n; i++)
        {
            pos[curIndex] = i;
            if(checkQueens(pos, curIndex))
            {
                totalNQueensCore(n, curIndex + 1, totalNumber, pos);
            }
        }
        
    }
    
    bool checkQueens(vector<int> &pos, int curIndex)
    {
        for(int i = 0; i < curIndex; i++)
        {
            if(pos[i] == pos[curIndex] || pos[i] - pos[curIndex] == i - curIndex || pos[i] - pos[curIndex] == curIndex - i)
            {
                return false;
            }
        }
        return true;
    }