[LeetCode] N-Queens II N皇后问题之二

简介:

Follow up for N-Queens problem.

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

这道题是之前那道 N-Queens N皇后问题 的延伸,说是延伸其实我觉得两者顺序应该颠倒一样,上一道题比这道题还要稍稍复杂一些,两者本质上没有啥区别,都是要用回溯法Backtracking来解,如果理解了之前那道题的思路,此题只要做很小的改动即可,不再需要求出具体的皇后的摆法,只需要每次生成一种解法时,计数器加一即可,代码如下:

class Solution {
public:
    int totalNQueens(int n) {
        int res = 0;
        vector<int> pos(n, -1);
        totalNQueensDFS(pos, 0, res);
        return res;
    }
    void totalNQueensDFS(vector<int> &pos, int row, int &res) {
        int n = pos.size();
        if (row == n) ++res;
        else {
            for (int col = 0; col < n; ++col) {
                if (isValid(pos, row, col)) {
                    pos[row] = col;
                    totalNQueensDFS(pos, row + 1, res);
                    pos[row] = -1;
                }
            }
        }
    }
    bool isValid(vector<int> &pos, int row, int col) {
        for (int i = 0; i < row; ++i) {
            if (col == pos[i] || abs(row - i) == abs(col - pos[i])) {
                return false;
            }
        }
        return true;
    }
};

本文转自博客园Grandyang的博客,原文链接:皇后问题之二[LeetCode] N-Queens II N,如需转载请自行联系原博主。

相关文章
|
机器学习/深度学习
LeetCode 51. N-Queens
n-queens难题是将n个皇后放置在n×n棋盘上的问题,使得没有两个皇后互相攻击。 给定整数n,返回n-queens拼图的所有不同解。
60 0
LeetCode 51. N-Queens
LeetCode - 51. N-Queens
51. N-Queens  Problem's Link  ---------------------------------------------------------------------------- Mean:  N-Queen问题.
843 0
LeetCode - 52. N-Queens II
52. N-Queens II  Problem's Link  ---------------------------------------------------------------------------- Mean:  略.
776 0
[LeetCode] N-Queens
The idea is to use backtracking. In fact, the code below uses DFS, which involves backtracking in a recursive manner.
734 0
[LeetCode] N-Queens II
If you have solved the N-Queens problem, this one can be solved in a similar manner. Starting from the first row, we try each of its columns.
607 0
|
22天前
|
机器学习/深度学习 算法
力扣刷题日常(一)
力扣刷题日常(一)
20 2