[Python3]数独计算器

学习 Python 之后发现似乎比 C 更容易掌握，于是也自己尝试写了一个小程序：

import sys
fo = open("sudoku_data.txt")
sudoku = [0, 0, 0, 0, 0, 0, 0, 0, 0]
for i in range(0,9):
    sudoku[i] = list(map(int, fo.readline().replace('\n', '')))
fo.close()
def canfill(i, j, k, sudoku): 
    '''当前行 列 填入的数字 数独'''
    for m in range(0, 9):
        if (sudoku[i][m] == k or sudoku[m][j] == k or sudoku[i // 3 * 3 + m // 3][j // 3 * 3 + m % 3] == k):
            return False
    return True
def solve(i, j, sudoku):
    if (i == 9):
        result = open("sudoku_result.txt",'w')
        for k in range(0, 9):
            for m in range(0, 9):
                result.write(str(sudoku[k][m]))
            result.write('\n')
        result.close()
        sys.exit()
    if (sudoku[i][j] != 0):
        solve(i if j < 8 else i + 1, (j + 1) % 9, sudoku)   
    for k in range(1, 10):
        if (canfill(i, j, k, sudoku)):
            sudoku[i][j] = k
            solve(i if j < 8 else i + 1, (j + 1) % 9, sudoku)     
            sudoku[i][j] = 0		#如果下一个数填写失败就让当前所填成功的数为0以避开上一个if的条件，进行下一个k的尝试
solve(0, 0, sudoku)

功能

在data中写入数独，然后运行程序，填写完毕的数独会在result中输出出来

递归

函数调用本身，从而实现回溯，尝试填入下一个k，最终解出答案。

if (sudoku[i][j] != 0):
        solve(i if j < 8 else i + 1, (j + 1) % 9, sudoku)   
    for k in range(1, 10):
        if (canfill(i, j, k, sudoku)):
            sudoku[i][j] = k
            solve(i if j < 8 else i + 1, (j + 1) % 9, sudoku)     
            sudoku[i][j] = 0

首先判断是否不为0，即区别空与已知条件。
从1到9的循环中尝试填入，满足条件可以调用自身进行下一个判断，如果1-9都不满足，就跳回上一层尝试。
由于结束程序时退出递归很麻烦，于是借用sys.exit()来强制退出。

三目运算

Python 中没有像 C 语言中一样直接使用 x ? a : b 这样的三元表达式，程序中的
solve(i if j < 8 else i + 1, (j + 1) % 9, sudoku)
其实等价于：

if (j < 8):
	solve(i, (j + 1) % 9, sudoku)
else:
	solve(i + 1, (j + 1) % 9, sudoku)

意思就是填写右边一个格子，如果已经是最右边，就令行数+1，列变为(8 + 1) % 9 = 0，从下一行第一个格子开始填写。

判断函数

def canfill(i, j, k, sudoku): 
    '''当前行 列 填入的数字 数独'''
    for m in range(0, 9):
        if (sudoku[i][m] == k or sudoku[m][j] == k or sudoku[i // 3 * 3 + m // 3][j // 3 * 3 + m % 3] == k):
            return False
    return True

前面的sudoku[i][m] == k or sudoku[m][j] == k很容易理解，之后是对于宫内数字是否重复的判断。

i // 3 * 3返回的结果是0, 3, 6，即每一宫的开始行，之后的j也同理。
m // 3 与 m % 3 可以让 m 在 0 - 8 的逐一遍历中返回
0 0
0 1
0 2
1 0
1 1
…
这样相加就可以表示出行列。