[leetcode] 52. N-Queens II

最新推荐文章于 2024-09-16 06:54:08 发布

TstsUgeg

最新推荐文章于 2024-09-16 06:54:08 发布

阅读量366

点赞数

分类专栏： leetcode 文章标签： leetcode backtracing

本文链接：https://blog.csdn.net/TstsUgeg/article/details/50008937

版权

leetcode 专栏收录该内容

254 篇文章 0 订阅

订阅专栏

这篇博客关注LeetCode的第52题，即N-Queens II问题。与第51题类似，但目标不是输出解决方案，而是计算不同解决方案的总数。该问题在算法领域被归类为Hard难度。

摘要由CSDN通过智能技术生成

Follow up for N-Queens problem.

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

这道题和第51题基本没有太大区别，题目难度为Hard。

如果需要详细解释请看第51题（传送门），具体代码：

class Solution {
public:
    int totalNQueens(int n) {
        vector<string> curRst(n, string(n, '.'));
        int cnt = 0;
        
        NQueens(curRst, 0, n, cnt);
        
        return cnt;
    }
    
    void NQueens(vector<string> &curRst, int row, int n, int &cnt) {
        if(row == n) {
            cnt++;
            return;
        }
        
        for(int i=0; i<n; i++) {
            bool flag = true;
            for(int j=0; j<row; j++) {
                if(curRst[j][i] == 'Q') {
                    flag = false;
                    break;
                }
                if((i+j-row >= 0) && (i+j-row <= n-1) && (curRst[j][i+j-row] == 'Q')) {
                    flag = false;
                    break;
                }
                if((i+row-j >= 0) && (i+row-j <= n-1) && (curRst[j][i+row-j] == 'Q')) {
                    flag = false;
                    break;
                }
            }
            
            if(flag) {
                curRst[row++][i] = 'Q';
                NQueens(curRst, row, n, cnt);
                curRst[--row][i] = '.';
            }
        }
    }
};