Python初心者がナンプレ問題作成プログラムを作ってみた-001（プログラムコード有り）

import random

# 関数の再定義（すべての必要な関数をインポート）
def is_valid(board, row, col, num):
    for i in range(9):
        if board[row][i] == num or board[i][col] == num:
            return False
    start_row, start_col = 3 * (row // 3), 3 * (col // 3)
    for i in range(start_row, start_row + 3):
        for j in range(start_col, start_col + 3):
            if board[i][j] == num:
                return False
    return True

def solve_sudoku(board):
    for row in range(9):
        for col in range(9):
            if board[row][col] == 0:
                for num in range(1, 10):
                    if is_valid(board, row, col, num):
                        board[row][col] = num
                        if solve_sudoku(board):
                            return True
                        board[row][col] = 0
                return False
    return True

def count_solutions(board):
    def solve(board):
        nonlocal count
        for row in range(9):
            for col in range(9):
                if board[row][col] == 0:
                    for num in range(1, 10):
                        if is_valid(board, row, col, num):
                            board[row][col] = num
                            solve(board)
                            board[row][col] = 0
                    return
        count += 1
    
    count = 0
    solve(board)
    return count

def remove_numbers_strict(board, holes):
    attempts = holes
    while attempts > 0:
        row, col = random.randint(0, 8), random.randint(0, 8)
        while board[row][col] == 0:
            row, col = random.randint(0, 8), random.randint(0, 8)
        backup = board[row][col]
        board[row][col] = 0
        
        board_copy = [row[:] for row in board]
        if count_solutions(board_copy) != 1:
            board[row][col] = backup
        else:
            attempts -= 1

    return board

def generate_complete_board_backtracking():
    def fill_board(board):
        for i in range(9):
            for j in range(9):
                if board[i][j] == 0:
                    random_nums = list(range(1, 10))
                    random.shuffle(random_nums)
                    for num in random_nums:
                        if is_valid(board, i, j, num):
                            board[i][j] = num
                            if fill_board(board):
                                return True
                            board[i][j] = 0
                    return False
        return True
    
    board = [[0] * 9 for _ in range(9)]
    fill_board(board)
    return board

def generate_sudoku_strict(difficulty):
    complete_board = generate_complete_board_backtracking()
    puzzle = remove_numbers_strict(complete_board, difficulty)
    return puzzle

def print_sudoku(board):
    print("+-------+-------+-------+")
    for i in range(9):
        row = "| "
        for j in range(9):
            if board[i][j] == 0:
                row += "  "
            else:
                row += str(board[i][j]) + " "
            if (j + 1) % 3 == 0:
                row += "| "
        print(row)
        if (i + 1) % 3 == 0:
            print("+-------+-------+-------+")

def solve_sudoku_and_print(board):
    def solve(board):
        for row in range(9):
            for col in range(9):
                if board[row][col] == 0:
                    for num in range(1, 10):
                        if is_valid(board, row, col, num):
                            board[row][col] = num
                            if solve(board):
                                return True
                            board[row][col] = 0
                    return False
        return True

    if solve(board):
        print_sudoku(board)
    else:
        print("解けない")

# 自動化プロセスの関数
def sudoku_solver_process(level):
    # 難易度に応じた穴の数を設定
    if level == "初級":
        holes = 30
    elif level == "中級":
        holes = 40
    elif level == "上級":
        holes = 50
    elif level == "難問":
        holes = 55
    else:
        print("無効なレベルです。レベルは「初級」「中級」「上級」「難問」から選んでください。")
        return

    print(f"\n{level}レベルのナンプレ問題:")
    puzzle = generate_sudoku_strict(holes)
    print_sudoku(puzzle)

    print(f"\n{level}レベルのナンプレ問題の解答:")
    solve_sudoku_and_print(puzzle)

# 例: 初級レベルのナンプレ問題を生成して解く
sudoku_solver_process("難問")