Leetcode -- N-Queens II

最新推荐文章于 2019-08-20 23:29:58 发布

原创最新推荐文章于 2019-08-20 23:29:58 发布 · 278 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

48 篇文章

订阅专栏

本文探讨了N皇后问题的回溯法解决策略，重点在于通过回溯法计算不同规模棋盘上的所有独特解决方案数量。

Follow up for N-Queens problem.

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

分析：

典型的回溯法题目。

class Solution {
public:
    int cnt=0;
    bool check(vector<int> pos,int r,int p)
    {
        for(int i=0;i<r;++i)
            if(pos[i]==p||abs(i-r)==abs(pos[i]-p)) return 0;
        return 1;
    }
    void fun(vector<int> pos,int r)
    {
        if(r==pos.size())
        {
            cnt++;
            return;
        }
        for(int i=0;i<pos.size();++i)
        {
            if(check(pos,r,i))
            {
                pos[r]=i;
                fun(pos,r+1);
            }
        }
    }
    int totalNQueens(int n) {
        vector<int> pos(n,0);
        fun(pos,0);
        return cnt;
    }
};