回溯算法——以八皇后问题为例

回溯算法

回溯算法也叫试探法，它是一种系统地搜索问题的解的方法。

基本思想

从一条路往前走，能进则进，不能进则退回来，换一条路再试。【能进则进】

一般步骤

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）