1. 引言
本文为介绍流行的数独游戏的系列文章中的第一篇。更具体地说,我们如何构建一个脚本来解决数独难题,本文的重点在于介绍用于构建数独求解器的回溯算法。
数独这个名字的由来来自日语短语suuji wa dokushin ni kagiru
,意思是“数字必须保持单一”。数独游戏的流行也源于其规则的简单性:数独游戏要求在 9 x 9 空间的网格上进行数字填写。在行和列中有 9 个“正方形”的格子block(由 3 x 3 个子单元格cell组成)。每一行、每一列、每一个block都需要填写数字 1-9,行、列、block内不得重复任何数字。
好的,知道了上述数独的规则,那么我们就来直接进入主体吧。 :)
2.描述数独九宫格
这一步主要为使用一组数字来初始化我们的九宫格。我们使用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
上述代码运行后,如果展示在拼图游戏中,他的样子大概如下:
3.寻找下一个空子单元格
函数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
4. 子单元格中设置值的合法性
函数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()
函数中描述的条件后成功插入新值的动态示例:
同时,若我们引入一个与九宫格数独上已经存在的值冲突的数值,那么此时该值将不会被插入。图例如下:
5. 实现回溯算法
下一个函数是我们整个解决方案的核心思想,这里引入了回溯思想,该算法的实现步骤如下:
- 搜索下一个空的子单元格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,直到出现死胡同;然后它回溯,直到重新迭代该过程。
6. 性能表现
上述我们的程序需要使用回溯算法来遍历每个单元格的许多潜在值。这当然不是最优的解法,可以预想到我们的解决方法的性能会很慢。
我们使用上述代码,来解决欧拉计划的第96题中的50道数独题目,运行时间为:
The execution time of above program is : 23.56185507774353 s
好吧,确实有点慢。我们后面再来开篇讲解数独算法的优化。
7. 总结
本文讲解了数独游戏的相关规则,以及如何在Python中利用回溯思想来一步一步解决数独问题,并给出了完整的实现。
您学废了吗?
参考
关注公众号《AI算法之道》,获取更多AI算法资讯。