深度优先搜索之八皇后问题
八皇后问题是一个古老而又著名的问题,是学习回溯算法的一个经典案例。我们不探究她的历史,只说解决方案
在8×8格的国际象棋上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问一共有多少种摆法。
class NQueen():
def __init__(self):
self.n = 8; #定义几个皇后
self.res = [] # 定义结果
def run(self):
self.bfs(resOfOne=[])
print(f"共有{
len(self.res)}种可能")
print(self.res)
def bfs(self,resOfOne):
if len(resOfOne) == self.n:
self.res.append(resOfOne[:])
for i in range(self.n):
# 确保不在同一行同一列
if i not in resOfOne:
# 确保不在同一斜线
if self.notINBias(i,resOfOne):
resOfOne.append(i)
self.bfs(resOfOne)
resOfOne.pop(-1)
def notINBias(self,i,resOfOne):
length = len(resOfOne)
for m,n in enumerate(resOfOne):
if length-m == i-n or length+i == n+m:
return False
return True
nqueen = NQueen()
nqueen.run()
这其实就是回溯,我们把这个问题划分成 8 个阶段,依次将 8 个棋子放到第一行、第二行、第三行……第八行。在放置的过程中,我们不停地检查当前放法,是否满足要求。如果满足,则跳到下一行继续放置棋子;如果不满足,那就再换一种放法,继续尝试。
实际上我认为回溯算法最难的是对于递归的理解
