这期内容当中小编将会给大家带来有关怎么在python中解决n皇后问题,文章内容丰富且以专业的角度为大家分析和叙述,阅读完这篇文章希望大家可以有所收获。
import copydef check(A,x,y): B=[] flag=True for i in range(len(A)): for j in range(len(A)): if A[i][j]==1: B.append([i,j]) for m in range(len(B)): p = B[m][0] q = B[m][1] if y == q or (x-p)==abs(y-q): flag=False return flagdef queen(n): A=[[0 for __ in range(n)] for _ in range(n)] answer=[] for _ in range(n): stack=[[0,_,A]] while stack: judge = 0 obj=stack.pop(-1) x=obj[0] y=obj[1] array=obj[2] flag=check(array,x,y) if not flag: while 1: if check(array, x, y): break else: if stack: b=stack.pop(-1) x=b[0] y=b[1] array=b[2] else: judge=1 break if judge==1: break array=copy.deepcopy(array) array[x][y]=1 for m in range(n): if m!=y and m!=y-1 and m!=y+1 and x+1<n : stack.append([x+1,m,array]) # print(array) for j in range(len(array[n-1])): if array[n-1][j]==1: answer.append(array) print(len(answer))queen(8)
answer中存放的就是最后所有的可行组合
当前解决的是8皇后问题
我的想法是用dfs 在每次搜索时 带上该次搜索需要摆放的位置 x,y,以及待摆放的棋盘 即[x,y,A]
这样不会导致所有的操作都在一个矩阵上进行
上述就是小编为大家分享的怎么在python中解决n皇后问题了,如果刚好有类似的疑惑,不妨参照上述分析进行理解。如果想知道更多相关知识,欢迎关注编程网行业资讯频道。