如何用单个for循环实现树的BFS遍历？验证层级检测伪代码正确性

原代码的核心bug：

• 初始状态崩溃：第一次遍历根节点时，PreviousNodeLevel是null，执行PreviousNodeLevel.insert(node)会直接报错——null根本没有插入方法。
• 提示重复触发：下一层的每个节点都满足node ∈ children(PreviousNodeLevel)的条件，会导致每遍历一个下一层节点就显示一次提示，而不是仅在进入新层时显示一次。
• 冗余逻辑分支：else if PreviousNodeLevel != null then PreviousNodeLevel ← null这个分支在BFS按层遍历的逻辑里根本不会触发，因为BFS是严格按层推进的，不会出现跳层遍历的情况。

修正后的单for循环BFS伪代码（满足你的需求） #

下面的代码保持了你要求的单个标注//(traverse BFS)的for循环，同时解决了上述问题：

// 初始化：根节点属于第一层
currentLevel ← { rootNode }
nextLevel ← empty set
newLevelFlag ← false

for node in tree do //(traverse BFS)
    // 进入新层时触发提示
    if newLevelFlag then
        display "已进入树的下一层"
        currentLevel ← nextLevel
        nextLevel ← empty set
        newLevelFlag ← false
    end if

    // 收集当前节点的子节点，为下一层遍历做准备
    nextLevel.addAll( children(node) )

    // 这里添加你处理当前节点的业务逻辑
    process(node)

    // 检查当前层是否遍历完成，标记下一轮进入新层
    if node == last element of currentLevel then
        newLevelFlag ← true
    end if
end for

修正逻辑说明 #

1. 用currentLevel记录当前正在遍历的层的所有节点，nextLevel提前收集下一层的所有子节点。
2. newLevelFlag作为标记，仅在新层的第一个节点遍历的时候显示一次提示，避免重复触发。
3. 每次遍历到当前层的最后一个节点时，将标记设为true，确保下一次循环只触发一次新层提示。

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