利用迭代协议实现自定义的递归深度控制

酷丽大大_5807

酷丽大大_5807

2026-06-24

745人浏览

原创

迭代协议本身不控制递归深度,但可用显式栈模拟递归并自主维护深度;如tree_depth_iterator用(stack, depth)元组避免调用栈,yield实现迭代,天然防recursionerror。

利用迭代协议实现自定义的递归深度控制

直接用迭代协议实现“递归深度控制”其实是个概念错位——迭代协议本身不处理调用栈或深度限制,它只定义对象如何被 for 循环或 next() 逐个访问。但你可以借助迭代器+显式状态管理,**绕过函数调用栈**,从而天然规避 RecursionError。这本质上是用迭代替代递归,并在过程中自主跟踪“当前深度”。

用迭代器模拟递归过程并记录深度

比如遍历一棵树时,你想在每层打印当前深度,同时防止爆栈。不靠递归函数调用,而是用一个栈(list 或 deque)存节点及其对应深度:

def tree_depth_iterator(root):
  if not root:
    return
  stack = [(root, 0)]  # (节点, 当前深度)
  while stack:
    node, depth = stack.pop()
    yield node, depth
    if node.right:
      stack.append((node.right, depth + 1))
    if node.left:
      stack.append((node.left, depth + 1))

使用方式:
for node, d in tree_depth_iterator(root):
  print(f"值: {node.val}, 深度: {d}")

关键点:
• 每个元素是 (node, depth),深度由你显式维护,不依赖调用栈
• yield 让它成为生成器,符合迭代协议,可被 for 遍历或 next() 逐步取值
• 不会触发 RecursionError,哪怕树深 10 万层

用自定义迭代器类封装深度逻辑

如果你需要更精细的状态控制(比如跳过某深度、限制最大遍历深度),可以写一个带参数的迭代器类:

LilyFM
LilyFM

一款AI音频处理工具,主要用于AI播客生成工具,一键将网页文章转换成播客,适合需要提升相关任务效率的用户。

下载

class BoundedDepthIterator:
  def __init__(self, root, max_depth=100):
    self.root = root
    self.max_depth = max_depth
    self.stack = [(root, 0)] if root else []
    self._exhausted = False

  def __iter__(self):
    return self

  def __next__(self):
    if not self.stack or self._exhausted:
      raise StopIteration
    node, depth = self.stack.pop()
    if depth > self.max_depth:
      return self.__next__()  # 跳过超深节点
    if node.right and depth + 1       self.stack.append((node.right, depth + 1))
    if node.left and depth + 1       self.stack.append((node.left, depth + 1))
    return node, depth

这样就实现了:主动限深、跳过无效分支、全程无函数递归调用。

为什么不用 setrecursionlimit?

sys.setrecursionlimit() 只是放宽 Python 的硬性拦截,但底层 C 栈空间仍有限。一旦真把深度设到几万,容易引发段错误或程序崩溃,尤其在嵌入式或容器环境中。
而基于迭代协议的方式:
• 所有状态存在 Python 对象(list/tuple)里,内存由垃圾回收管理
• 深度变量只是整数,不增加栈帧
• 可随时中断、暂停、检查、修改深度逻辑
• 天然支持协程、异步迭代等扩展场景

配合 itertools 做流式截断

若你已有生成器,还想按深度做分批处理,可用 itertools.islice 或 takewhile:

from itertools import takewhile

# 只取深度 ≤ 5 的所有节点
shallow_nodes = takewhile(lambda x: x[1] for node, d in shallow_nodes:
  process(node)

或者用 islice 控制最多取前 N 层的结果,避免一次性加载全量。

相关专题

更多
js获取数组长度的方法
js获取数组长度的方法

在js中,可以利用array对象的length属性来获取数组长度,该属性可设置或返回数组中元素的数目,只需要使用“array.length”语句即可返回表示数组对象的元素个数的数值,也就是长度值。php中文网还提供JavaScript数组的相关下载、相关课程等内容,供大家免费下载使用。

2023.06.20

4526

5

js刷新当前页面
js刷新当前页面

js刷新当前页面的方法:1、reload方法,该方法强迫浏览器刷新当前页面,语法为“location.reload([bForceGet]) ”;2、replace方法,该方法通过指定URL替换当前缓存在历史里(客户端)的项目,因此当使用replace方法之后,不能通过“前进”和“后退”来访问已经被替换的URL,语法为“location.replace(URL) ”。php中文网为大家带来了js刷新当前页面的相关知识、以及相关文章等内容

2023.07.04

1129

3

js四舍五入
js四舍五入

js四舍五入的方法:1、tofixed方法,可把 Number 四舍五入为指定小数位数的数字;2、round() 方法,可把一个数字舍入为最接近的整数。php中文网为大家带来了js四舍五入的相关知识、以及相关文章等内容

2023.07.04

4444

6

js删除节点的方法
js删除节点的方法

js删除节点的方法有:1、removeChild()方法,用于从父节点中移除指定的子节点,它需要两个参数,第一个参数是要删除的子节点,第二个参数是父节点;2、parentNode.removeChild()方法,可以直接通过父节点调用来删除子节点;3、remove()方法,可以直接删除节点,而无需指定父节点;4、innerHTML属性,用于删除节点的内容。

2023.09.01

900

4

JavaScript转义字符
JavaScript转义字符

JavaScript中的转义字符是反斜杠和引号,可以在字符串中表示特殊字符或改变字符的含义。本专题为大家提供转义字符相关的文章、下载、课程内容,供大家免费下载体验。

2023.09.04

1796

5

js生成随机数的方法
js生成随机数的方法

js生成随机数的方法有:1、使用random函数生成0-1之间的随机数;2、使用random函数和特定范围来生成随机整数;3、使用random函数和round函数生成0-99之间的随机整数;4、使用random函数和其他函数生成更复杂的随机数;5、使用random函数和其他函数生成范围内的随机小数;6、使用random函数和其他函数生成范围内的随机整数或小数。

2023.09.04

3245

4

如何启用JavaScript
如何启用JavaScript

JavaScript启用方法有内联脚本、内部脚本、外部脚本和异步加载。详细介绍:1、内联脚本是将JavaScript代码直接嵌入到HTML标签中;2、内部脚本是将JavaScript代码放置在HTML文件的`<script>`标签中;3、外部脚本是将JavaScript代码放置在一个独立的文件;4、外部脚本是将JavaScript代码放置在一个独立的文件。

2023.09.12

4213

6

Js中Symbol类详解
Js中Symbol类详解

javascript中的Symbol数据类型是一种基本数据类型,用于表示独一无二的值。Symbol的特点:1、独一无二,每个Symbol值都是唯一的,不会与其他任何值相等;2、不可变性,Symbol值一旦创建,就不能修改或者重新赋值;3、隐藏性,Symbol值不会被隐式转换为其他类型;4、无法枚举,Symbol值作为对象的属性名时,默认是不可枚举的。

2023.09.20

2720

5

java访问控制修饰符介绍
java访问控制修饰符介绍

java访问控制修饰符有四种,分别是public、protected、private、默认访问修饰符。详细介绍:1、public,public是最宽松的访问控制修饰符,被修饰的类、方法和变量可以被任何其他类访问,当一个类、方法或变量被声明为public时,它们可以在任何地方被访问,无论是同一个包中的类还是不同包中的类;2、protected修饰符等等。

2023.09.20

888

7

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习