生成器函数天然适配递归结构,因其yield机制精准对应“暂停→处理子问题→恢复组合”的递归本质;改写只需三步:保留组合逻辑、递归调用前加yield、外层用yield from展开;运行时通过for循环或send驱动,无需手动模拟栈帧。

生成器函数天然适配递归结构
递归的本质是“暂停当前状态→处理子问题→恢复并组合结果”,而生成器的 yield 正好对应这个暂停点。不需要手动建栈、压参、设 flag,也不用拆解 return 位置。只要把原递归调用替换成 yield,函数就自动变成可中断、可恢复的状态机。
关键改写三步走
以树的后序遍历为例(需先处理左右子树再访问根):
- 原递归写法中,return left_result + right_result + [root.val] 这类组合逻辑,改写时保留不变
- 所有递归调用如 postorder(root.left) 替换为 yield postorder(root.left)
- 在生成器外部用 yield from 或手动迭代展开子生成器,确保值逐层上冒
运行时用简单循环驱动生成器
生成器本身不执行,需要外层控制流来“拉取”结果。常见模式是:
- 用 for value in generator(...): 直接消费最终值
- 对嵌套 yield,用 yield from sub_generator() 委托,避免手动 while-next 循环
- 若需中间状态或提前终止,可调用 gen.send(value) 向 yield 表达式注入数据,实现双向通信
比纯栈模拟更贴近原意图
传统栈模拟法要把参数、局部变量、返回地址全打包进栈帧,代码膨胀且易错。生成器方案把上下文隐式保留在闭包和执行暂停点中,逻辑分支、条件判断、多路递归调用都无需额外标记。你写的还是“像递归”的代码,只是运行时不依赖调用栈。

















