题目链接:受伤的皇后
一道经典的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))