[Leetcode][Python]52: N-Queens II

最新推荐文章于 2022-11-15 16:32:31 发布

aaron7909

最新推荐文章于 2022-11-15 16:32:31 发布

阅读量164

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：http://www.cnblogs.com/Dabay/p/4373035.html

本文介绍了一种解决N皇后问题的算法，通过深度优先搜索和递归的方式，计算在N×N的棋盘上放置N个皇后，使得任意两个皇后不处于同一行、列或对角线上的方案总数。文章详细展示了算法的实现过程，包括检查皇后放置位置的合法性以及递归搜索所有可能的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -*- coding: utf8 -*-
'''
__author__ = 'dabay.wang@gmail.com'

52: N-Queens II
https://oj.leetcode.com/problems/n-queens-ii/

Follow up for N-Queens problem.
Now, instead outputting board configurations, return the total number of distinct solutions.

===Comments by Dabay===
不知道和N Queen相比有没有简单很多的方法。我这里的解法思路和N Queen一样的。

一个一个放皇后，知道能放下最后一个皇后，解法+1。
放第k个皇后的时候，在第k行中找位置，先看列被占用没有，然后往左上和右上看斜线被占用没有。
'''

class Solution:
    # @return an integer
    def totalNQueens(self, n):
        def check_up(r, c, board):
            for row in xrange(r):
                if board[row][c] == 'Q':
                    return False
            else:
                return True

        def check_upleft(r, c, board):
            row = r - 1
            column = c - 1
            while row>=0 and column>=0:
                if board[row][column] == 'Q':
                    return False
                row = row - 1
                column = column - 1
            else:
                return True

        def check_upright(r, c, board):
            row = r - 1
            column = c + 1
            while row>=0 and column<len(board):
                if board[row][column] == 'Q':
                    return False
                row = row - 1
                column = column + 1
            else:
                return True

        def DFS(board, queens, res):
            if queens == 0:
                res[0] = res[0] + 1
                return
            r = len(board) - queens
            for c in xrange(len(board)):
                if not check_up(r, c, board) or not check_upleft(r, c, board) or not check_upright(r, c, board):
                    continue
                else:
                    board[r][c] = 'Q'
                    DFS(board, queens-1, res)
                    board[r][c] = '.'

        board = [['.'] * n for _ in xrange(n)]
        #print board
        queens = n
        res = [0]
        DFS(board, queens, res)
        return res[0]


def main():
    sol = Solution()
    print sol.totalNQueens(5)


if __name__ == "__main__":
    import time
    start = time.clock()
    main()
    print "%s sec" % (time.clock() - start)