[leetcode]052-N-Queens II[递归回溯]

最新推荐文章于 2021-05-03 15:58:56 发布

原创最新推荐文章于 2021-05-03 15:58:56 发布 · 259 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

Leetcode 专栏收录该内容

53 篇文章

订阅专栏

本文介绍了解决N皇后问题的第二种方法，专注于计算所有可能的解决方案数量，而非保存每种解决方案。通过深度优先搜索(DFS)算法，递归地检查每个位置是否能放置皇后，并确保没有冲突。文章提供了详细的C++实现代码，展示了如何验证每一行、列及对角线的有效性。

1.原题

https://leetcode.com/problems/n-queens-ii/

2. 思路

这题和上道题51思路是一样的，只不过，搜索到结果后不需要保存，

只需要计数就可以了。

3. 源码

class Solution {
public:
    int totalNQueens(int n) {
        dim = n;
        vector<string> mv(n, string(n, '.'));
        dfs(mv, 0);
        return num;
    }
    
    void dfs(vector<string> &mv, int row)
    {
        if (row == dim)  //right answer
        {
            num++;
            return;
        }

         for (int i = 0; i < dim; i++)
        {
            if (isValid(mv, row, i)) //current position is valid
            {
                mv[row][i] = 'Q';
                dfs(mv, row+1);     //process next row by recursion
                mv[row][i] = '.';
            }
        }
        return;
    }

     bool isValid(vector<string> &mv, int x, int y)
    {
        for (int i = 0; i < x; i++ ) // validate column data, here no need for row
        {                            // because we process row up to down
            if (mv[i][y] != '.')
                return false;
        }
        for (int i=x-1, j = y-1; i > -1 && j > -1; i--,j--) // master diagonal
        {
            if (mv[i][j] != '.')
                return false;
        }
        for (int i = x-1, j = y+1; i>-1 && j<dim; i--,j++) // subdiagnonal
        {
            if (mv[i][j] != '.')
                return false;
        }
        return true;
    }
private:
    int num{0};
    int dim;
};