Python竞赛编程Fast I/O优化：解决NZEC读取输入输出空值问题求助

• 循环内重复初始化io.BytesIO对象：os.read(0, os.fstat(0).st_size)的作用是一次性读取标准输入的全量字节，每轮循环重新执行这段初始化逻辑时，会生成一个全新的流对象，不仅流指针始终停在内存缓冲区的起始位置，无法连续按顺序读取后续输入，还会因为系统文件偏移被反复重置，最终因指针位置错乱读到空内容，输出空行。
• 未规范处理字节解码与换行符：零散调用decode()时未统一处理首尾空白、换行符偏移的问题，进一步放大了读取异常。

输入侧优化逻辑 #

• 全量输入预加载仅执行一次：程序启动时直接通过系统调用把所有标准输入读到内存，全程不做逐行系统读请求，把I/O overhead降到最低。
• 两种可选实现等级：
  • 通用流读取版本：适合需要逐行动态读取输入的场景，保证流指针自动偏移到正确位置：

    import io, os
    # 全量预加载仅执行1次，绝对不能放进循环
    input_buf = io.BytesIO(os.read(0, os.fstat(0).st_size))
    # 封装标准化读行方法，自动解码、去除首尾换行/空白
    def read_str():
        return input_buf.readline().decode().strip()
    

  • 极致性能版本：适合超大规模数据集（百万行级输入），一次性把全量输入解码后切分成列表，直接通过指针索引取值，比流读取速度高30%以上：

    import os
    input_lines = os.read(0, os.fstat(0).st_size).decode().splitlines()
    ptr = 0
    def read_str():
        nonlocal ptr
        cur_val = input_lines[ptr].strip()
        ptr += 1
        return cur_val
    

输出侧优化逻辑 #

• 禁止循环内反复调用print()：单次print会触发一次系统写调用，当输出量达到万行级时，反复的系统调用开销会直接导致程序超时。
• 统一用缓冲写输出：把所有需要打印的内容先存入列表，最后拼接成完整字符串一次性写出：

  import sys
  out_buf = []
  # 业务逻辑中把要输出的内容追加到out_buf
  out_buf.append(str(计算结果))
  # 全部处理完成后一次性写出
  sys.stdout.write('\n'.join(out_buf))
  

对应多组输入测试场景的完整可运行版本：

import io, os, sys

# 初始化仅执行一次
input_buf = io.BytesIO(os.read(0, os.fstat(0).st_size))
test_case_cnt = int(input_buf.readline().decode().strip())

for _ in range(test_case_cnt):
    cur_input = input_buf.readline().decode().strip()
    sys.stdout.write(f"{cur_input}\n")

内容的提问来源于stack exchange，提问作者DRV