数独挖洞算法 python代码

以下是一个基于Python的数独挖洞算法示例代码：

import random

def generate_puzzle():
    """
    生成数独谜题
    """
    # 初始化一个空的9x9数独网格
    puzzle = [[0 for x in range(9)] for y in range(9)]
    # 生成第一行
    puzzle[0] = random.sample(range(1, 10), 9)
    # 逐行生成数独谜题
    for row in range(1, 9):
        for col in range(9):
            # 根据已有数字生成当前行
            available_nums = [i for i in range(1, 10) if i not in puzzle[row]]
            # 根据已有数字生成当前列
            for i in range(row):
                if puzzle[i][col] in available_nums:
                    available_nums.remove(puzzle[i][col])
            # 根据已有数字生成当前九宫格
            square_row = (row // 3) * 3
            square_col = (col // 3) * 3
            for i in range(square_row, square_row + 3):
                for j in range(square_col, square_col + 3):
                    if puzzle[i][j] in available_nums:
                        available_nums.remove(puzzle[i][j])
            # 随机从可用数字中选择一个填入当前格
            puzzle[row][col] = random.choice(available_nums)
    return puzzle

def remove_numbers(puzzle, holes):
    """
    挖洞，移除数独中的数字
    """
    # 复制一份数独谜题
    puzzle_copy = [row[:] for row in puzzle]
    # 初始化已挖洞的格子数量
    holes_count = 0
    # 随机选择格子挖洞，直到达到指定数量的洞
    while holes_count < holes:
        # 随机选择一行和一列
        row = random.randint(0, 8)
        col = random.randint(0, 8)
        # 如果当前格子未挖洞，将其挖洞
        if puzzle_copy[row][col] != 0:
            puzzle_copy[row][col] = 0
            holes_count += 1
    return puzzle_copy

使用示例：

# 生成一个数独谜题
puzzle = generate_puzzle()
# 挖洞，移除30个数字
puzzle_with_holes = remove_numbers(puzzle, 30)
# 输出数独谜题
for row in puzzle_with_holes:
    print(row)

这个示例代码生成一个数独谜题，然后使用 `remove_numbers` 函数挖掉指定数量的数字。你可以根据需要修改 `generate_puzzle` 和 `remove_numbers` 函数中的参数来生成不同的数独谜题。