蓝桥云课——受伤的皇后，n皇后 Python （深度优先遍历）

 题目链接：受伤的皇后

        一道经典的DFS题目，有助于你深入了解所谓深度优先遍历

DFS相应模板（具体情况具体分析,有时需要用到栈）：

def DFS(node):
    # 相应代码
    if 语句:  # 用于终止递归
        return
    # 相应代码
    for 循环:  # 用于循环递归
        # 相应代码
        DFS(node)
        # 相应代码

---

在解决这个问题之前我们先解决所谓n皇后问题

按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。

先输入数据：

chess[i]表示第几行第几列放了皇后，chess[4]  = 7 表示第三行第七列有皇后

n = int(input())
chess = [0 for i in range(n)]
ans = 0

再写DFS函数：

def DFS(row):
    global n
    global ans
    if row == n:  # 终止条件
        ans += 1  # 方案+1
        return

    for i in range(n):  # 对这从左到右一行一个一个放皇后
        if check(row, i):
            chess[row] = i
            DFS(row + 1)  # 对下一行放皇后
            chess[row] = 0  # 再将其归0，因为下一步已经是另一种情况
    return ans

最后补充check函数：

        chess[i] == y 表示该点纵向是否存在皇后，横向不用判断，因为你是从横向开始放置的

        chess[i] + i == x + y 表示右上到左下45°，画个图能够发现其对角线上的所有点的（x,y）位置之和都相同

        i - chess[i] == x - y 表示左上到右下45°，画个图能够发现其对角线上的所有点的（x,y）位置之差都相同

def check(x, y):
    for i in range(x):
        if chess[i] == y or chess[i] + i == x + y or i - chess[i] == x - y:
            return 0
    return 1

n皇后完整代码：

n = int(input())
chess = [0 for i in range(n)]
ans = 0


def check(x, y):
    for i in range(x):
        if chess[i] == y or chess[i] + i == x + y or i - chess[i] == x - y:
            return 0
    return 1


def DFS(row):
    global n
    global ans
    if row == n:
        ans += 1
        return

    for i in range(n):
        if check(row, i):
            chess[row] = i
            DFS(row + 1)
            chess[row] = 0
    return ans


print(DFS(0))

---

对于本题受伤的皇后只需要把check函数中的条件修改即可：

def check(x, y):
    for i in range(x):
        if chess[i] == y:
            return 0
        if i + chess[i] == x + y and (x - i) ** 2 < 9:  # 行号的差值的平方<9
            return 0
        if i - chess[i] == x - y and (x - i) ** 2 < 9:  # 行号的差值的平方<9
            return 0
    return 1

受伤的皇后完整代码：

n = int(input())
chess = [0 for i in range(n)]
ans = 0


def check(x, y):
    for i in range(x):
        if chess[i] == y:
            return 0
        if i + chess[i] == x + y and (x - i) ** 2 < 9:
            return 0
        if i - chess[i] == x - y and (x - i) ** 2 < 9:
            return 0
    return 1


def DFS(row):
    global n
    if row == n:
        global ans
        ans += 1
        return

    for i in range(n):
        if check(row, i):
            chess[row] = i
            DFS(row + 1)
            chess[row] = 0
    return ans


print(DFS(0))