一个自动生成数独题目的Python脚本

from PIL import Image, ImageDraw, ImageFont
import random

# 定义一个 9x9 的二维数组来表示数独棋盘
board = [[0 for _ in range(9)] for _ in range(9)]

# 定义一个函数来检查某个数字在给定的位置是否合法
def is_valid(num, row, col):
    # 检查所在行是否有相同数字
    for i in range(9):
        if board[row][i] == num:
            return False
    # 检查所在列是否有相同数字
    for i in range(9):
        if board[i][col] == num:
            return False
    # 检查所在 3x3 小方格是否有相同数字
    start_row = row // 3 * 3
    start_col = col // 3 * 3
    for i in range(3):
        for j in range(3):
            if board[start_row + i][start_col + j] == num:
                return False
    return True

# 定义一个函数来生成数独谜题
def generate_sudoku():
    for row in range(9):
        for col in range(9):
            num_list = list(range(1, 10))
            random.shuffle(num_list)
            for num in num_list:
                if is_valid(num, row, col):
                    board[row][col] = num
                    if solve_sudoku():  # 尝试完成剩余部分，如果成功则继续
                        return True
                    board[row][col] = 0  # 回溯
    return False

# 定义一个函数来解决数独（用于检查生成的谜题是否可解）
def solve_sudoku():
    for row in range(9):
        for col in range(9):
            if board[row][col] == 0:
                for num in range(1, 10):
                    if is_valid(num, row, col):
                        board[row][col] = num
                        if solve_sudoku():
                            return True
                        board[row][col] = 0
                return False
    return True

# 生成完整的数独答案
generate_sudoku()
for row in board:
    print(row)


# 随机挖空一些数字来生成题目
num_empty_cells = random.randint(30, 50)  # 控制挖空的数量，例如 30 到 50 个
for _ in range(num_empty_cells):
    row = random.randint(0, 8)
    col = random.randint(0, 8)
    if board[row][col]!= 0:
        board[row][col] = 0

# 绘制图片
cell_size = 50
margin = 10
image_width = 9 * cell_size + 2 * margin
image_height = 9 * cell_size + 2 * margin

image = Image.new('RGB', (image_width, image_height), (255, 255, 255))
draw = ImageDraw.Draw(image)

font = ImageFont.truetype("PingFang.ttc", 30)  # 请确保您有 PingFang 字体文件

# 绘制横线和竖线
for i in range(10):
    if i % 3 == 0:
        line_width = 3
    else:
        line_width = 1
    draw.line([(margin + i * cell_size, margin), (margin + i * cell_size, image_height - margin)], fill=(0, 0, 0), width=line_width)
    draw.line([(margin, margin + i * cell_size), (image_width - margin, margin + i * cell_size)], fill=(0, 0, 0), width=line_width)

# 填写数字
for row in range(9):
    for col in range(9):
        if board[row][col]!= 0:
            x = margin + col * cell_size + cell_size // 2
            y = margin + row * cell_size + cell_size // 2
            draw.text((x, y), str(board[row][col]), fill=(0, 0, 0), font=font, anchor="mm")

image.save('sudoku.png')