51. N-Queens

这一题的写法是模仿discussion中的Java的。总的来说分为以下几个部分,初始化board的时候把所有的点的值都初始化为'.',在遍历的时候按照列索引,判断列中行坐标的点是否可以放置,直到列索引达到最后,这样可以确定一种答案

class Solution(object):
    def solveNQueens(self, n):
        """
        :type n: int
        :rtype: List[List[str]]
        """
        board = [['.' for i in range(n)] for i in range(n)]
        res = []
        self.dfs(0, board, res)
        return res
        
    def dfs(self, curcol, board, res):
        if curcol == len(board):
            res.append(self.construct(board))
            return
        for i in range(len(board)):
            if self.check(board, i, curcol):
                board[i][curcol] = 'Q'
                self.dfs(curcol+1, board, res)
                board[i][curcol] = '.'       
    
    def check(self, board, x, y):
        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
    def construct(self, board):
        res = []
        for i in range(len(board)):
            s = ''
            for j in range(len(board)):
                s += board[i][j]
            res.append(s)
        return res
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容