python matrix取元素_用Python解数独[3]：求总值域

最新推荐文章于 2021-11-25 15:34:10 发布

weixin_39743369

最新推荐文章于 2021-11-25 15:34:10 发布

阅读量144

点赞数

文章标签： Python 数独算法值域列表交集

本文链接：https://blog.csdn.net/weixin_39743369/article/details/111625832

版权

用Python解数独[0]
用Python解数独[1]：求每个单元格的行值域
用Python解数独[2]：求列值域和九宫格值域
用Python解数独[3]：求总值域
用Python解数独[4]：缩减值域
用Python解数独[5]：检测唯一值缩减值域
用Python解数独[6]：递归获得最终答案
用Python解数独[7]：递归（完结篇）

[2][1] 解决数独问题基本思路

4、求每个单元格的总值域

上回我们说到，我们已经获得了一个单元格的行值域、列值域和九宫格值域，可以考虑通过三个值域来获得这个单元格的总值域了。这个问题可以抽象为有lista , listb和listc，求三个列表的交集。

求lista, listb, listc三个列表的交集，其实可以分解为两步

求 lista 和 listb 的交集 tempa
求 tempa 和 listc 的交集 tempb

我们这样就把一个求三个列表交集的问题，简化为求两个列表交集，这样看起来简单了不少，代码：

def inersection(lista,listb):
    tempa = []
    for i in lista:
        for j in listb:
	    if i == j:
	        tempa.append(i)
    return tempa

代码基本逻辑就是，双重遍历lista和listb，如果lista中的元素i与listb中任一元素相等，那么将i放到tempa中

那么求三个列表交集的函数就有了多种写法：

第一种

def intersection(lista,listb):
    tempa = []
    for i in lista:
        for j in listb:
	    if i == j:
	        tempa.append(i)
    return tempa

def intersection_triple(lista,listb,listc):
    tempa = intersection(lista,listb)
    tempb = intersection(tempa,listc)
    return tempb

第二种

def inersection(lista,listb,listc):
    tempa = []
    tempb = []
    for i in lista:
        for j in listb:
            if i == j:
                tempa.append(i)
    for k in listc:
        for l in tempa:
            if k == l:
                tempb.append(k)
    return tempb

我们用row_value_range表示行值域列表，col_value_range表示行值域列表，matrix_value_range表示九宫格值域列表，但它们都是二维数组，因此需要找到同一个单元格对应的行值域、列值域和九宫格值域，然后用inersection函数取交集，获得每个单元格的总值域，然后把这些总值域按照一定的规律组成一个新的二维数组，我们用代码展示一下：

def totalValueRange(soduku):
    row_value_range = rowValueRange(soduku)
    col_value_range = colValueRange(soduku)
    matrix_value_range = matrixValueRange(soduku)
    total_value_range = [[] for i in range(9)]
    for i in range(9):
        for j in range(9):
            total_value_range[i].append(inersection(row_value_range[i][j],col_value_range[i][j],matrix_value_range[i][j]))
    return total_value_range

我们整理一下现有的代码

import copy
import time

# 求每一行单元格行值域
def valueRange(row):
    temp = copy.deepcopy(row)
    row_value_range = list(range(1,10))
    for i in row:
        if i == '':
            continue
        else:
            row_value_range.remove(i)
    for j in range(9):
        if temp[j] == '':
            temp[j] = row_value_range
        else:
            temp[j] = [temp[j]]
    return temp

# 求数独每个单元格行值域
def rowValueRange(soduku):
    row_value_range = []
    for row in soduku:
        row_value_range.append(valueRange(row))
    return row_value_range

# 求数独每个单元格列值域
def colValueRange(soduku):
    soduku_invert = list(zip(*soduku))
    temp = rowValueRange(soduku_invert)
    s_column_vrange = list(zip(*temp))
    return s_column_vrange

# 将一个数独数组转化为数独九宫格数组
def matrix_invert(lista):
    listb = [[] for i in range(9)]
    for i in range(9):
        for j in range(9):
            k = i//3
            l = j//3
            m = k*3 + l
            listb[m].append(lista[i][j])
    return listb

# 求数独每个单元格九宫格值域
def matrixValueRange(soduku):
    matrix = matrix_invert(soduku)
    temp = rowValueRange(matrix)
    matrix_vrange = matrix_invert(temp)
    return matrix_vrange

# 三个列表求交集函数
def inersection(lista,listb,listc):
    tempa = []
    tempb = []
    for i in lista:
        for j in listb:
            if i == j:
                tempa.append(i)
    for k in listc:
        for l in tempa:
            if k == l:
                tempb.append(k)
    return tempb

#  求数独每个单元格总值域
def totalValueRange(soduku):
    row_value_range = rowValueRange(soduku)
    col_value_range = colValueRange(soduku)
    matrix_value_range = matrixValueRange(soduku)
    total_value_range = [[] for i in range(9)]
    for i in range(9):
        for j in range(9):
            total_value_range[i].append(inersection(row_value_range[i][j],col_value_range[i][j],matrix_value_range[i][j]))
    return total_value_range

if __name__ == '__main__':
    t1 = time.time()

    soduku = [[] for i in range(9)]

    soduku[0] = ['',8,9,1,'',3,'','','']
    soduku[1] = ['',2,7,4,'','',8,'','']
    soduku[2] = ['','','',5,2,'','','','']
    soduku[3] = ['',7,6,9,'','',5,'',8]
    soduku[4] = [8,'',3,6,'',1,2,'','']
    soduku[5] = ['',4,5,3,8,7,1,6,'']
    soduku[6] = [7,'',8,2,1,4,3,9,5]
    soduku[7] = [9,3,1,7,6,5,4,'',2]
    soduku[8] = [4,'',2,'','',9,'','',6]
    
    for i in totalValueRange(soduku):
        print(i)

    print("代码执行完毕，用时{}秒".format(round(time.time() - t1,2)))

执行代码，发现报错了......

PS D:python0624-soduku> python zhihutest.py
Traceback (most recent call last):
  File "zhihutest.py", line 92, in <module>
    for i in totalValueRange(soduku):
  File "zhihutest.py", line 69, in totalValueRange
    col_value_range = colValueRange(soduku)
  File "zhihutest.py", line 30, in colValueRange
    temp = rowValueRange(soduku_invert)
  File "zhihutest.py", line 24, in rowValueRange
    row_value_range.append(valueRange(row))
  File "zhihutest.py", line 15, in valueRange
    temp[j] = row_value_range
TypeError: 'tuple' object does not support item assignment

报错了怎么办呢？调试代码debug吧，乖乖地把报错信息里的行数都打上断点，然后执行debug

在行号左侧点击鼠标右键就是“打断点”，代码执行到这一行时会中断，点击下一步就会继续执行代码
根据报错信息，是类型错误，也就是tuple（元组）类别不支持索引。猜测我们在执行某个函数时，应该输入一个列表，结果我们输入的是一个元组，导致错误；
在执行30行代码前，我们发现问题了，soduku_invert的每个元素应该是列表，结果却是元组；

那么为什么会这样呢？我们看29行代码，这个和zip函数有关，也是我之前疏忽了

soduku_invert = list(zip(*soduku))

zip函数根据soduku的每一行，生成soduku的每一列，以每列为元素组成一个zip对象，我们可以用list()方法将zip对象转化为列表，但此时列表里的每一个元素是元组而不是列表！

soduku_invert = zip(*soduku)
print(soduku_invert)

PS D:python0624-soduku> python test6.py
<zip object at 0x02EE70D0>

soduku_invert = list(zip(*soduku))
print(soduku_invert)

PS D:python0624-soduku> python test6.py
[('', '', '', '', 8, '', 7, 9, 4), (8, 2, '', 7, '', 4, '', 3, ''), (9, 7, '', 6, 3, 5, 8,
1, 2), (1, 4, 5, 9, 6, 3, 2, 7, ''), ('', '', 2, '', '', 8, 1, 6, ''), (3, '', '', '', 1, 7, 4, 5, 9), 
('', 8, '', 5, 2, 1, 3, 4, ''), ('', '', '', '', '', 6, 9, '', ''), ('', '', '', 8, '', '', 5, 2, 6)]

所以29行代码需要略微修改，将soduku_invert里的每一个元素用list()方法转化为数组，31行也要修改

soduku_invert = [list(i) for i in list(zip(*soduku))]
temp = rowValueRange(soduku_invert)
s_column_vrange = [list(i) for i in list(zip(*temp))]

将29行、31行修改后，我们重新执行代码

PS D:python0624-soduku> python zhihutest.py
[[5, 6], [8], [9], [1], [7], [3], [6, 7], [2, 4, 5, 7], [4, 7]]
[[1, 3, 5, 6], [2], [7], [4], [9], [6], [8], [1, 3, 5], [1, 3, 9]]
[[1, 3, 6], [1, 6], [4], [5], [2], [6, 8], [6, 7, 9], [1, 3, 4, 7], [1, 3, 4, 7, 9]]
[[1, 2], [7], [6], [9], [4], [2], [5], [3, 4], [8]]
[[8], [9], [3], [6], [4, 5], [1], [2], [4, 7], [4, 7, 9]]
[[2], [4], [5], [3], [8], [7], [1], [6], [9]]
[[7], [6], [8], [2], [1], [4], [3], [9], [5]]
[[9], [3], [1], [7], [6], [5], [4], [8], [2]]
[[4], [5], [2], [8], [3], [9], [7], [1, 7, 8], [6]]
代码执行完毕，用时0.0秒

稍微整理一下，就是下图，看着有什么问题？

比如第1行第7/8/9列，明明可以排除掉7了，为什么值域里还有呢？因为我们只是根据已知数字获得了值域，在整理之后，直接获知了第1行第5列的答案是7，但是这个信息是整理后获知的，并不是一开始的已知信息，因此我们得到的结果，还是包含很多“无效信息”的。如果我们根据新获得的答案，继续缩小值域就有可能获得更多答案，那么应该如何用代码表示，又应该在何时结束循环呢？

预知后事如何，且听下回分解。

下一篇链接：用Python解数独[4]：缩减值域

weixin_39743369

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫