
本文介绍一种时间复杂度为 o(n + m) 的高效方案:通过哈希查找一次性定位目标文件,避免嵌套遍历;使用 python 的 `os.listdir()` 与文件流安全追加,支持存在性校验与容错处理。
在处理大量按字母排序的 .txt 文件(如 aaa.txt, aab.txt, …, zzz.txt)时,若需将另一组非连续命名的文件(如 ant.txt, cat.txt, lion.txt)内容追加到对应同名文件中,暴力双重循环(O(n×m))显然低效且不必要——我们无需“人工跳转索引”,而应借助操作系统级目录扫描 + 哈希映射实现常数时间查找。
核心优化思路是:只遍历一次源文件集(第二组),对每个文件名,直接构造目标路径并检查其是否存在。这依赖于文件系统对 os.path.isfile() 的高效实现(通常为 O(1) 元数据查询),整体时间复杂度降为 O(m),其中 m 是第二组文件数量;空间开销几乎为零,无需预加载内容或维护额外索引。
以下是生产就绪的 Python 实现:
import os
import sys
def append_second_to_first(target_dir: str, source_dir: str) -> None:
"""
将 source_dir 中每个 .txt 文件的内容,追加到 target_dir 中同名文件末尾。
若 target_dir 中不存在对应文件,则跳过(可选行为,见下方说明)。
"""
# 确保路径存在且为目录
if not (os.path.isdir(target_dir) and os.path.isdir(source_dir)):
raise ValueError("Both target_dir and source_dir must be valid directories.")
for filename in os.listdir(source_dir):
# 仅处理 .txt 文件(可按需扩展)
if not filename.lower().endswith('.txt'):
continue
source_path = os.path.join(source_dir, filename)
if not os.path.isfile(source_path):
continue
target_path = os.path.join(target_dir, filename)
# 关键优化:仅当目标文件存在时才追加(满足题设“仅合并已存在文件”语义)
if os.path.isfile(target_path):
try:
with open(source_path, 'r', encoding='utf-8') as src_f, \
open(target_path, 'a', encoding='utf-8') as dst_f:
# 逐行追加,避免内存爆炸(适合大文件)
for line in src_f:
dst_f.write(line)
print(f"✓ Appended '{filename}' to '{target_path}'")
except (IOError, OSError) as e:
print(f"⚠ Failed to process '{filename}': {e}")
else:
print(f"ℹ Skipping '{filename}': no matching file in target directory")
if __name__ == "__main__":
if len(sys.argv) != 3:
print("Usage: python appendfiles.py ")
sys.exit(1)
append_second_to_first(sys.argv[1], sys.argv[2]) ✅ 关键优势说明:
- 无嵌套循环:摒弃 for src in sources: for tgt in targets: 的 O(n×m) 模式;
- 存在性优先判断:os.path.isfile(target_path) 在打开前完成校验,避免 FileNotFoundError;
- 编码健壮性:显式指定 utf-8 编码,防止中文等非 ASCII 内容乱码;
- 大文件友好:逐行读写而非 readlines() 全量加载,内存占用恒定;
- 错误隔离:单个文件失败不影响其余处理,并输出清晰日志。
⚠️ 注意事项:
- 若需自动创建缺失的目标文件(例如将 bat.txt 复制到 target_dir),请删除 if os.path.isfile(target_path): 判断,并将 'a' 模式改为 'a+' 或先用 'w' 创建空文件;
- 确保运行用户对两个目录具有读取(source)和写入(target)权限;
- 生产环境建议添加日志模块(如 logging)替代 print,便于审计;
- 如需并发加速(数百+文件),可结合 concurrent.futures.ThreadPoolExecutor,但注意文件 I/O 通常受磁盘带宽限制,线程数不宜过多(4–8 为佳)。
该方案直击问题本质:利用文件系统天然的“名称即键”特性,以最简逻辑达成最优性能,兼顾可读性、健壮性与可维护性。










