面试题 08.12. 八皇后 -对角线的特征和标记方法

特别鸣谢：来自夸夸群的 醉笑陪公看落花@知乎，王不懂不懂@知乎，
感谢醉笑陪公看落花@知乎 倾囊相授，感谢小伙伴们督促学习，一起进步

相关文章:

• 棋盘类 剑指 Offer 47. 礼物的最大价值 两种解法-带备忘录的递归实现DFS和 动态规划

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

注意：本题相对原题做了扩展

示例:

输入：4
输出：[[".Q…","…Q",“Q…”,"…Q."],["…Q.",“Q…”,"…Q",".Q…"]]
解释: 4 皇后问题存在如下两个不同的解法。
[
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],

["…Q.", // 解法 2
“Q…”,
“…Q”,
“.Q…”]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/eight-queens-lcci

分析

• N个皇后放N*N的棋盘,必然一行只能放一个
• 放一个皇后之后, 该皇后所在的行,列,左对角,右对角方向的位置不能使用
• 第一行有N种放法,每一种尝试一下,就尝试完了全体情况
• 递归的时候就考虑, 当前行放置皇后之后,标记不能使用的位置,如果所有皇后都放入棋盘,说明这个方案是可行的.
• 对角线特征: (i,j)所在左对角线: i1+j1 = i+j = D1,右对角线 j1-i1 = j-i=D2

下面草图演示了 N=4 时, 其中的三种放置起点.

标记不能用的点:

# 标记成不能用的
# # 1. i+j =D1		# A
# # 2. j-i =D2		# B
# # 3. i			# 放第k个皇后的时候,用k控制即可
# # 4. j			# C

代码 - 递归,给第k个皇后找位置- 通过

class Solution:
    def solveNQueens(self, n: int):
        N = n
        A = [True]*2*N
        B = [True]*2*N
        C = [True]*N
        ans = []
        def D(k,hj): # 给第k个皇后找位置
            if k==N:
                ans.append([''.join(i) for i in hj])
                return
            for j in range(N):
                if A[k+j] and B[j-k] and C[j]:
                    hj[k][j]='Q'
                    A[k+j] = B[j-k] = C[j] = False
                    D(k+1,hj)
                    A[k+j] = B[j-k] = C[j] = True
                    hj[k][j]='.'
        D(0, [['.']*N for i in range(N)])
        return ans