回溯算法
回溯算法也叫试探法,它是一种系统地搜索问题的解的方法。
基本思想
从一条路往前走,能进则进,不能进则退回来,换一条路再试。【能进则进】
一般步骤
1 针对所给问题,定义问题的解空间,它至少包含问题的一个(最优)解。
2 确定易于搜索的解空间结构,使得能用回溯法方便地搜索整个解空间 。
3 以深度优先的方式搜索解空间,并且在搜索过程中用剪枝函数避免无效搜索。
4、利用限界函数避免移动到不可能产生解的子空间。
问题的解空间通常是在搜索问题的解的过程中动态产生的,这是回溯算法的一个重要特性。
确定了解空间的组织结构后,回溯法就从开始结点(根结点)出发,以深度优先的方式搜索整个解空间。这个开始结点就成为一个活结点,同时也成为当前的扩展结点。在当前的扩展结点处,搜索向纵深方向移至一个新结点。这个新结点就成为一个新的活结点,并成为当前扩展结点。如果在当前的扩展结点处不能再向纵深方向移动,则当前扩展结点就成为死结点。此时,应往回移动(回溯)至最近的一个活结点处,并使这个活结点成为当前的扩展结点。回溯法即以这种工作方式递归地在解空间中搜索,直至找到所要求的解或解空间中已没有活结点时为止。
八皇后问题
在 8×8 格的国际象棋上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法。
解题思路
path记录已经遍历的每行皇后的合法位置;
dfs遍历每行,判断每个位置是否和已经遍历的结果冲突,如果不冲突就继续往下遍历;
判断冲突的条件:同行、同列、主对角线、副对角线。
class Solution:
def solveNQueens(self, n: int) -> List[List[str]]:
def conflict(j, path):
ij = len(path)
for ik, k in enumerate(path):
if k == j or k-ik == j-ij or k+ik == j+ij:
return True
return False
def dfs(i, path):
if len(path) == n:
result.append(path[:])
return
for j in range(n):
if not conflict(j, path):
dfs(j, path + [j])
result = []
for i in range(n):
dfs(i, [i])
row = '.' * n
ans = []
for res in result:
tmp = []
for i in res:
tmp.append(row[:i] + 'Q' + row[i+1:])
ans.append(tmp[:])
return ans
链接:https://leetcode-cn.com/problems/eight-queens-lcci/solution/python-mian-shi-ti-0812-ba-huang-hou-by-9bsqz/
来源:力扣(LeetCode)