做一些有趣的小东西程序员python热爱者

Python 解数独(Sudoku)

2018-01-18  本文已影响99人  _Phine

闲来有了用python解数独的想法,但由于对复杂些的算法仍是一窍不通,最终算是用简单算法实现了出来。

相关简介:

1.使用的算法很常规,很好理解,有点类似深度优先搜索算法。
2.解常规难度的数独耗时约50~150 ms,但对网上的超难数独尚不能短时间内解出。 - -0
3.输入数独数据要么要input一行行手输,要么在程序中替换default_data数据,总之没有图形界面,输入有点不方便。

后续可能会继续优化性能,也可能改成图形界面~

使用方法:

主要是输入介绍,我设定了两种,游戏开始前可选择方式“1”或“2”:

***Method 1: Modify the default sudoku***
***Method 2: Through input method***

Please select the data input method:1 or 2
default_data = [[1 ,0 ,6 ,2 ,0 ,0 ,0 ,0 ,0 ],
                [0 ,0 ,0 ,4 ,0 ,0 ,8 ,2 ,0 ],
                [2 ,0 ,0 ,0 ,0 ,5 ,0 ,0 ,0 ],
                [0 ,8 ,0 ,0 ,4 ,0 ,0 ,0 ,7 ],
                [0 ,0 ,0 ,6 ,0 ,3 ,0 ,0 ,0 ],
                [5 ,0 ,0 ,0 ,1 ,0 ,0 ,4 ,0 ],
                [0 ,0 ,0 ,9 ,0 ,0 ,0 ,0 ,0 ],
                [0 ,3 ,9 ,0 ,0 ,4 ,0 ,0 ,0 ],
                [0 ,0 ,0 ,0 ,0 ,2 ,9 ,0 ,5 ]]
***Method 1: Modify the default sudoku***
***Method 2: Through input method***

Please select the data input method:1 or 2
2
Enter 1th line: 1 0 6 2 0 0 0 0 0
Enter 2th line: 0 0 0 4 0 0 8 2 0
Enter 3th line: 2 0 0 0 0 5 0 0 0
Enter 4th line: 0 8 0 0 4 0 0 0 7
Enter 5th line: 0 0 0 6 0 3 0 0 0
Enter 6th line: 5 0 0 0 1 0 0 4 0
Enter 7th line: 0 0 0 9 0 0 0 0 0
Enter 8th line: 0 3 9 0 0 4 0 0 0
Enter 9th line: 0 0 0 0 0 2 9 0 5

当你输入完毕会打印你输入的Sudoku,然后问你是否确认,输入Y确认后即可得到答案,N则重新输入:

Your data:
 1  0  6  2  0  0  0  0  0 
 0  0  0  4  0  0  8  2  0 
 2  0  0  0  0  5  0  0  0 
 0  8  0  0  4  0  0  0  7 
 0  0  0  6  0  3  0  0  0 
 5  0  0  0  1  0  0  4  0 
 0  0  0  9  0  0  0  0  0 
 0  3  9  0  0  4  0  0  0 
 0  0  0  0  0  2  9  0  5 
Confirm? Y or N

原理介绍

主要函数有两个:

def judge(data, x, y, num): 
    if data[y].count(num) > 0:   #行判断
        #print('error1')
        return False

    for col in range(9):   #列判断
        if data[col][x] == num:
            #print('error2')
            return False

    for a in range(3):   #九宫格判断
        for b in range(3):
            if data[a+3*(y//3)][b+3*(x//3)] == num:
                #print('error3')
                return False
    return True
def build_data_list(data):   
    data_list = []
    for y in range(9):
        for x in range(9):
            if data[y][x] == 0:
                data_list.append([(x, y), [1, 2, 3, 4, 5, 6, 7, 8, 9]])   #列表中有坐标信息以及备选的数字
    return data_list

为每一个空处都建立一个备选列表,而fill_num函数则是在递归所有的空,循环其中的备选列表。备选列表是可以在每一步前被筛选掉一部分,从而提高速度的,不过这一部分工作还没想好怎么做。

fill_num函数代码如下:

def fill_num(data, data_list, start): 
    if start < len(data_list):
        one = data_list[start]
        for num in one[1]:
            if judge(data, one[0][0], one[0][1], num):
                data[one[0][1]][one[0][0]] = num
                tem_data = fill_num(data, data_list, start+1)   #递归部分
                if tem_data != None:
                    return tem_data   #当递归返回值非空时才传给上层
        data[one[0][1]][one[0][0]] = 0   #当for loops结束时需要执行清零操作
    else:
        return data

fill_num函数中有坑:

if tem_data != None:
    return tem_data   #当递归返回值非空时才传给上层

因此增加这部分代码,避免某函数for loops结束,返回值为None时,引起外层函数连续return,中断操作。

if judge(data, one[0][0], one[0][1], num):   #judge为True就允许赋值
    data[one[0][1]][one[0][0]] = num

解决方案为,一旦for loops能完整执行,表明前面一定填错了,因此要将每一个错误的填数清零

data[one[0][1]][one[0][0]] = 0   #当for loops结束时需要执行清零操作

解决了这俩坑就没啥大问题了。

文件放在这里了,enjoy yourself~
Sudoku v2

Sudoku v2
上一篇下一篇

猜你喜欢

热点阅读