zoukankan      html  css  js  c++  java
  • LeetCode N-Queens

    LeetCode解题之N-Queens


    原题

    经典的八皇后问题的普通情况,用Python如何来高速地解决呢?

    8-queens

    注意点:

    • 皇后用”Q”表示,空白用”.”表示

    样例:

    输入: n = 4

    输出:

    [  ['.Q..', 
        '...Q', 
        'Q...', 
        '..Q.'], 
    
       ['..Q.', 
        'Q...', 
        '...Q', 
        '.Q..']]

    解题思路

    用三个数组来表示列、正反对角线的占用情况。一行行的遍历,假设没有冲突就把对应的位置置为占用,继续处理下一行,并记录改行的皇后放在了哪一列,当皇后都放完后。依据记录的列号来拼出结果。进行回溯时要把占用的位置还回去。

    对角线位置的计算要小心(尤其是反对角线),能够把顶点带进去计算验证一下。

    AC源代码

    class Solution(object):
        def solveNQueens(self, n):
            """
            :type n: int
            :rtype: List[List[str]]
            """
            self.col = [False] * n
            self.diag = [False] * (2 * n)
            self.anti_diag = [False] * (2 * n)
            self.result = []
            self.recursive(0, n, [])
            return self.result
    
        def recursive(self, row, n, column):
            if row == n:
                self.result.append(list(map(lambda x: '.' * x + 'Q' + '.' * (n - 1 - x), column)))
            else:
                for i in range(n):
                    if not self.col[i] and not self.diag[row + i] and not self.anti_diag[n - i + row]:
                        self.col[i] = self.diag[row + i] = self.anti_diag[n - i + row] = True
                        self.recursive(row + 1, n, column + [i])
                        self.col[i] = self.diag[row + i] = self.anti_diag[n - i + row] = False
    
    
    if __name__ == "__main__":
        print(Solution().solveNQueens(5))

    欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源代码。

  • 相关阅读:
    排队打饭 sdut 2443【最简单的贪心法应用举例】
    sdut 2445 小学数学
    最终排名 sdut 2446
    sort函数用法
    sdut1598 周游列国【简单模拟题】
    sdut 2441 屠夫与狼
    男生女生分配
    qsort函数详解
    test1.A[【dfs简单题】
    sdut 2449走迷宫【最简单的dfs应用】
  • 原文地址:https://www.cnblogs.com/jzssuanfa/p/7050735.html
Copyright © 2011-2022 走看看