Python解数独算法

array = [["5", "3", ".", ".", "7", ".", ".", ".", "."],
         ["6", ".", ".", "1", "9", "5", ".", ".", "."],
         [".", "9", "8", ".", ".", ".", ".", "6", "."],
         ["8", ".", ".", ".", "6", ".", ".", ".", "3"],
         ["4", ".", ".", "8", ".", "3", ".", ".", "1"],
         ["7", ".", ".", ".", "2", ".", ".", ".", "6"],
         [".", "6", ".", ".", ".", ".", "2", "8", "."],
         [".", ".", ".", "4", "1", "9", ".", ".", "5"],
         [".", ".", ".", ".", "8", ".", ".", "7", "9"]]


def dfs(pos):
    global valid
    if pos == len(spaces):
        valid = True
        return

    i, j = spaces[pos]
    for digit in range(9):
        if line[i][digit] == column[j][digit] == block[i // 3][j // 3][digit] is False:
            line[i][digit] = column[j][digit] = block[i // 3][j // 3][digit] = True
            array[i][j] = str(digit + 1)
            dfs(pos + 1)
            line[i][digit] = column[j][digit] = block[i // 3][j // 3][digit] = False
        if valid:
            return


line = [[False] * 9 for i in range(9)]
column = [[False] * 9 for i in range(9)]
block = [[[False] * 9 for i in range(3)] for j in range(3)]
valid = False
spaces = []

for i in range(9):
    for j in range(9):
        if array[i][j] == '.':
            spaces.append([i, j])
        else:
            digit = int(array[i][j]) - 1
            line[i][digit] = column[j][digit] = block[i // 3][j // 3][digit] = True

dfs(0)
print(array)