[LeetCode]题解（python）：052-N-Queens II

---

题目来源

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

Follow up for N-Queens problem.

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

---

题意分析

---

Input: n

Output: n of number of the result

Conditions:n 皇后问题，但是返回数目

---

题目思路

---

上题采用递归方法，这次采用非递归方法做，主要是弄清楚非递归什么时候回溯即可。同上题，采用一维数组存储第i行的位置，然后每一次判断条件也是跟上题一样（一维表示不会同行，board[i] != board[j]表示同列，abs（i- j）!= abs（list[i] - list[j]）表示不会同斜线）。

回溯的方法是，如果在本行没找到把row -= 1，然后把board[row] = -1(注意row已经更新了)，col = board[row] + 1（从新的一行的已有结果的下一列开始找）

 

row的判断：

当row等于0并且还需要回溯时，即可终止寻找

当row等于n-1时，此时说明有一个答案（前提是这一行已经找到了）

---

AC代码（Python）

---

 

__author__ = 'YE'

class Solution(object):
    def totalNQueens(self, n):
        """
        :type n: int
        :rtype: int
        """
        def check(k, j):
            for i in range(k):
                if board[i] == j or abs(k - i) == abs(board[i] - j):
                    return False
            return True

        board = [-1 for i in range(n)]

        row = 0
        col = 0
        count = 0


        while row < n:
            while col < n:
                if check(row, col):
                    board[row] = col
                    col = 0
                    break
                else:
                    col += 1
            if board[row] == -1:
                if row == 0:
                    break
                else:
                    row -= 1
                    col = board[row] + 1
                    board[row] = -1
                    continue
            if row == n - 1:
                count += 1
                col = board[row] + 1
                board[row] = -1
                continue
            row += 1
        return count



n = 4
print(Solution().totalNQueens(n))