在疫情防控的关键时刻,在家闲来无事,就尝试着在手机上玩数独游戏。但是解起来比较让人头疼,一不小心填错就要从头再来,费时耗力。(但是对于锻炼思维是极好的,哈哈哈哈哈哈哈哈嗝…)
为了能够快速解决数独问题,顺便温习温习一个寒假没有摸过的python和算法,所以就计划自己编程来实现数独的自动填充。
采用的思想是**回溯**。
通俗一点说就是从头到尾一个一个试,有点类似于深度优先搜索,当进行到某一个位置不能再向下进行时,就退回一步继续这个过程。
当从头到尾遍历完成,也就是当深度搜索一直到达了最后一个节点,所填入的数据均合乎规则,则此时填入的这一系列数字组合就是原始数独的解,此时即完成了数独的求解。
算法描述如下:
算法:
输入:数独阵。9行9列,没有数字的填0。
输出:填写完整的数独阵。
步骤:
1. 根据输入阵input_array,计算得到标志阵flag, 这是一个二值阵,1表示有原始数据,0表示没有。
2. 根据输入阵input_array,计算得到每一个未填写数据的位置处,不允许再填入的数据,用一个三维列表存贮,记为invalid_init。
3. 再根据flag阵,计算得到每一个0位置处对应的回溯位置坐标值,用于后面回溯,得到loc阵。
4. 从(0,0)开始循环遍历输入阵:
5. While i < 9:
6. J = 0
7. While j < 9:
8. If flag[I,j] == 1:
9. 继续下一个点,j++
10. Else:
11. 找出(i,j)位置处允许填入的有效数字
12. 循环遍历有效数字:
13. 填入一个数,判断是否合法
14. 若合法,则跳出循环,j++,继续下一个位置
15. 否则,将当前填入的值加入该位置对应的非法值列表中。
16. 循环结束后判断循环结束的原因,
17. 若是遍历完了所有有效数字后结束,说明该位置已经无法填值
18. 进行回溯
19. 回溯时将该位置的非法值列表还原到最初
20. I++
21. 循环结束后,即可完成数独填写。
具体代码实现如下(采用的语言为python):
import numpy as np
import copy
# 得到一个二维numpy阵,元素为0或者1,0代表原始输入无数据,1代表原始输入已填入数字
def getflag_array(input_array):
return np.int64(input_array > 0)
# 得到输入矩阵每一个位置上在回溯时应该跳转的位置,类似于父节点的位置
def getloclist(flag):
m,n = np.shape(flag)
N = [[[] for _ in range(n)] for _ in range(m)] # 存储回溯时的路径
temp = [-1,-1]
flg = True
for i in range(m):
for j in range(n):
if flag[i,j] == 1:
N[i][j] = [-1,-1]
elif flg:
N[i][j] = [i,j] # 第一个0所在位置处,在回溯时还是自身位置
temp = [i,j]
flg = False
else:
N[i][j] = temp # 其余均为上一个0出现的位置
temp = [i,j]
return N
# 得到矩阵每一个未填的位置处对应的 ★★★不可以填入★★★ 的数字。
def get_invalid_mat(input_array):
m,n = np.shape(input_array