美文网首页
52. N-Queens II

52. N-Queens II

作者: April63 | 来源:发表于2018-06-28 15:14 被阅读0次

    这一题跟1一样,比1简单一点,只需要个数就可以了

    class Solution(object):
        def totalNQueens(self, n):
            """
            :type n: int
            :rtype: int
            """
            board = [['.' for i in range(n)] for j in range(n)]
            res = [0]
            self.dfs(board, 0, res)
            return res[0]
        def dfs(self, board, curcol, res):
            if curcol == len(board):
                res[0] += 1
            for i in range(len(board)):
                if self.check(i, curcol, board):
                    board[i][curcol] = 'Q'
                    self.dfs(board, curcol+1, res)
                    board[i][curcol] = '.'
        def check(self, x, y, board):
            for i in range(len(board)):
                for j in range(y):
                    if board[i][j] == 'Q' and ((x+j == y+i) or (x+y == i+j) or (x == i)):
                        return False
            return True
    

    相关文章

      网友评论

          本文标题:52. N-Queens II

          本文链接:https://www.haomeiwen.com/subject/lzigyftx.html