使用Python进行数独求解（一）

本文为介绍流行的数独游戏的系列文章中的第一篇。更具体地说，我们如何构建一个脚本来解决数独难题，本文的重点在于介绍用于构建数独求解器的回溯算法。

数独这个名字的由来来自日语短语suuji wa dokushin ni kagiru，意思是“数字必须保持单一”。数独游戏的流行也源于其规则的简单性：数独游戏要求在 9 x 9 空间的网格上进行数字填写。在行和列中有 9 个“正方形”的格子block(由 3 x 3 个子单元格cell组成)。每一行、每一列、每一个block都需要填写数字 1-9，行、列、block内不得重复任何数字。

好的，知道了上述数独的规则，那么我们就来直接进入主体吧。 ：）

这一步主要为使用一组数字来初始化我们的九宫格。我们使用setBoard() 函数将输入转换为适合我们后续操作的数据类型。我们使用以下约定：

空的单元格cell初始化为默认值0。维持数独谜题数字值的数据类型是一个 9x9 大小的二维列表。

这里我们的输入是一个多行字符串，我们将其处理成二维列表的形式。样例代码如下：

# Initialize a 2-D list with initial values described by the problem. 
# Returns board
def setBoard():
    board = list()
    sudokuBoard = '''
    200080300
	060070084
	030500209
	000105408
	000000000
	402706000
	301007040
	720040060
	004010003
	'''
    rows = sudokuBoard.split('n')
    for row in rows:
        column = list()
        for character in row:
            digit = int(character)
            column.append(digit)
        board.append(column)
    return board

上述代码运行后，如果展示在拼图游戏中，他的样子大概如下：

函数findEmpty() 函数的操作更加简单：对初始化后的九宫格作为参数传递，然后该遍历该九宫格中每一个子单元格cell，直到找到返回的第一个空的子单元格。如果没有找到空的子单元格，这表明我们的问题已解决，因此它返回None。
样例代码如下：

# Find next empty space in Sudoku board and return dimensions
def findEmpty(board):
    for row in range(9):
        for col in range(9):
            if board[row][col] == 0 :
                return row,col
    return None

函数isValid()的操作是确认设定的数字是否是九宫格子单元格的有效选项。为了使设置的值满足数独九宫格的要求，该值的设置需满足以下条件：

同一行的任何子单元格cell都不应包含该数字同一列的任何子单元格cell都不应包含该数字该子单元格cell所在的block不应该包含该数字

如果我们设定的值满足以上所有条件，该函数返回True，否则返回False。代码如下：

# Check whether a specific number can be used for specific dimensions
def isValid(board, num, pos):
    row, col = pos
    # Check if all row elements include this number
    for j in range(9):
        if board[row][j] == num:
            return False
    # Check if all column elements include this number
    for i in range(9):
        if board[i][col] == num:
            return False
    # Check if the number is already included in the block
    rowBlockStart = 3* (row // 3)
    colBlockStart = 3* (col // 3)

    rowBlockEnd = rowBlockStart + 3
    colBlockEnd = colBlockStart + 3
    for i in range(rowBlockStart, rowBlockEnd):
        for j in range(colBlockStart, colBlockEnd):
            if board[i][j] == num:
                return False

    return True

以下是通过isValid() 函数中描述的条件后成功插入新值的动态示例：

同时，若我们引入一个与九宫格数独上已经存在的值冲突的数值，那么此时该值将不会被插入。图例如下：

下一个函数是我们整个解决方案的核心思想，这里引入了回溯思想，该算法的实现步骤如下：

搜索下一个空的子单元格cell。如果没有找到，那么我们已经解决了这个难题；如果没有，则转到第 2 步。通过迭代数字1-9 来猜测正确的数字，并参考已经确定的数字来检查它是否是一个合法的数字。如果找到一个有效的数字，此时递归调用solve() 函数并猜测下一个空的子单元格cell。如果数字1-9均无效，则将将子单元格cell的值重置为 0，并继续迭代以查找下一个有效数字。

# Solve Sudoku using backtracking
def solve(board):
    blank = findEmpty(board)
    if not blank:
        return True
    else:
        row, col = blank
    for i in range(1,10):
        if isValid(board, i, blank):
            board[row][col] = i

            if solve(board):
                return True

            board[row][col] = 0
    return False

由于递归理解起来不是那么直观，不妨让我们尝试使用动态来将整个过程形象化：

正如我们在上面的示例中看到的那样，该算法用有效数字填充空子单元格cell，直到出现死胡同；然后它回溯，直到重新迭代该过程。

上述我们的程序需要使用回溯算法来遍历每个单元格的许多潜在值。这当然不是最优的解法，可以预想到我们的解决方法的性能会很慢。
我们使用上述代码，来解决欧拉计划的第96题中的50道数独题目，运行时间为：

The execution time of above program is : 23.56185507774353 s

好吧，确实有点慢。我们后面再来开篇讲解数独算法的优化。

本文讲解了数独游戏的相关规则，以及如何在Python中利用回溯思想来一步一步解决数独问题，并给出了完整的实现。

您学废了吗？

参考

关注公众号《AI算法之道》，获取更多AI算法资讯。