Python学习日志2021-01-29

八皇后

设计一种算法，打印 N 皇后在 N × N 棋盘上的各种摆法，其中每个皇后都不同行、不同列，也不在对角线上。这里的“对角线”指的是所有的对角线，不只是平分整个棋盘的那两条对角线。

基于集合的回溯

为了判断一个位置所在的列和两条斜线上是否已经有皇后，使用三个集合 $\text{columns},{\text{diagonals}}_1,{\text{diagonals}}_2$分别记录每一列以及两个方向的每条斜线上是否有皇后。
列的表示法很直观，一共有 N 列，每一列的下标范围从 0 到 N−1，使用列的下标即可明确表示每一列。
如何表示两个方向的斜线呢？对于每个方向的斜线，需要找到斜线上的每个位置的行下标与列下标之间的关系。
方向一的斜线为从左上到右下方向，同一条斜线上的每个位置满足行下标与列下标之差相等，例如 (0,0) 和 (3,3) 在同一条方向一的斜线上。因此使用行下标与列下标之差即可明确表示每一条方向一的斜线。
方向二的斜线为从右上到左下方向，同一条斜线上的每个位置满足行下标与列下标之和相等，例如 (3,0) 和 (1,2) 在同一条方向二的斜线上。因此使用行下标与列下标之和即可明确表示每一条方向二的斜线。

class Solution:
    def solveNQueens(self, n: int) -> List[List[str]]:
        def backtrack(row :int):
            if row==n:
                board = generateBoard()
                solutions.append(board)
            else:
                for i in range(n):
                    if i in columns or row-i in diagonals1 or row+i in diagonals2:
                        continue
                    queens[row]=i
                    columns.add(i)
                    diagonals1.add(row-i)
                    diagonals2.add(row+i)
                    backtrack(row+1)
                    columns.remove(i)
                    diagonals1.remove(row-i)
                    diagonals2.remove(row+i)
        
        def generateBoard():
            board = list()
            for i in range(n):
                rows[queens[i]]="Q"
                board.append("".join(rows))
                rows[queens[i]]="."
            return board
        
        solutions = list()
        columns = set()
        queens = [-1]*n
        rows = ['.']*n
        diagonals1 = set()
        diagonals2 =set()
        backtrack(0)

        return solutions