51. N-Queens

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u012033124/article/details/80695353

https://leetcode.com/problems/n-queens/description/
n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。皇后的走法:可以到达横向、纵向和斜线(45度,135度)的任意距离。
给定一个整数 n,返回所有不同的 n 皇后问题的解决方案。
每一种解法包含一个明确的 n 皇后问题的棋子放置方案,该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

思路:回溯。从第一行开始填,每次对0~n-1列遍历,检查每个位置是否合法,若合法则填入,并递归进入第二行。

用到了python的多变量for循环,用zip函数。注意循环的终点以范围最短的为准:

>>> for i,j,k in zip(range(9), range(1,9), range(2,11)):  #每次对三个变量分别循环
...    print(i,j,k)
... 
0 1 2
1 2 3
2 3 4
3 4 5
4 5 6
5 6 7
6 7 8
7 8 9  #j先到达终点,所有变量一起结束

此外由于回溯棋盘时,string的元素不能修改,因此用一个二维字符列表,而不是一维string列表来表示棋盘。而最终结果要求一个一维string列表,因此得到的结果还需要转换,将字符列表转为string,用join函数:
https://blog.csdn.net/u013810296/article/details/57082163

class Solution:
    def solveNQueens(self, n):
        """
        :type n: int
        :rtype: List[List[str]]
        """
        record = []
        board = [['.'] * n for _ in range(n)]  #注意二维列表的创建方式
        self.solve(record, board, 0, n)
        return record

    def solve(self, record, board, row, n):
        if row == n:  #得到一个结果
            res = []  #记录一个结果
            for i in range(n):
                res.append(''.join(board[i]))  #将棋盘每行转为一个string填入结果
            record.append(res)
        else:
            for col in range(n):
                if self.isValid(board, row, col, n):  #若合法则填入
                    board[row][col] = 'Q'
                    self.solve(record, board, row + 1, n)  #递归进入下一行
                    board[row][col] = '.'  #还原现场,回溯

    def isValid(self, board, row, col, n):  #判断row,col位置是否合法
        for i in range(n):
            if board[i][col] == 'Q':
                return False
        for i, j in zip(range(row - 1, -1, -1), range(col - 1, -1, -1)):
            if board[i][j] == 'Q':
                return False
        for i, j in zip(range(row - 1, -1, -1), range(col + 1, n)):
            if board[i][j] == 'Q':
                return False
        return True
阅读更多
相关热词
换一批

没有更多推荐了,返回首页