🔴 真的调一次 C

🔴 这一节的 Python 题会**在运行时调 gcc 现编一个 .so，再用 ctypes 加载进来**——真的跨语言调用，不是假装。判题机需要 gcc、一个可写的临时目录，以及允许 Python 起子进程。⚠️ 这条路线**一个答案都不是耗时**。耗时跟机器、跟负载、跟编译器都有关，换台机器数字就变了。**能当答案的只有"走了多少步"和"两边结果一不一样"。**

import ctypes, os, subprocess, tempfile

def build_c(src, name="m"):
    """把一段 C 现场编成 .so，交回加载好的库。
       判题机需要 gcc 和一个可写的临时目录。"""
    d = tempfile.mkdtemp()
    c_path = os.path.join(d, name + ".c")
    so_path = os.path.join(d, name + ".so")
    with open(c_path, "w") as f:
        f.write(src)
    subprocess.run(["gcc", "-shared", "-fPIC", "-O2", "-o", so_path, c_path],
                   check=True, capture_output=True)
    return ctypes.CDLL(so_path)

C_SRC = r"""
int add(int a, int b) { return a + b; }
long sum_to(long n) { long s = 0; for (long i = 1; i <= n; i++) s += i; return s; }
"""

lib = build_c(C_SRC)

# 每个要调的函数都得先说清楚：参数是什么、交回什么
lib.add.argtypes = [ctypes.c_int, ctypes.c_int]
lib.add.restype = ctypes.c_int
lib.sum_to.argtypes = [ctypes.c_long]
lib.sum_to.restype = ctypes.c_long

c_result = lib.sum_to(100)
py_result = sum(range(1, 101))

print(f"{lib.add(3, 4)}/{c_result}/{int(c_result == py_result)}")