diff --git a/problems/二叉树的统一迭代法.md b/problems/二叉树的统一迭代法.md index a1ac9dd2..412aec8c 100644 --- a/problems/二叉树的统一迭代法.md +++ b/problems/二叉树的统一迭代法.md @@ -31,8 +31,8 @@ * 方法一:**就是要处理的节点放入栈之后,紧接着放入一个空指针作为标记。** 这种方法可以叫做`空指针标记法`。 -* 方法二:**当一个节点被`pop()`后,把该节点放入一个`Set`中,表示该节点被处理过了,下次再处理这个节点时,直接收割。** -这种方法可以叫做`Set标记法`,样例代码见下文`Python Set标记法`。 方法二更容易理解,在面试中更容易写出来。 +* 方法二:**加一个 `boolean` 值跟随每个节点,`false` (默认值) 表示需要为该节点和它的左右儿子安排在栈中的位次,`true` 表示该节点的位次之前已经安排过了,可以收割节点了。** +这种方法可以叫做`boolean 标记法`,样例代码见下文`C++ 和 Python 的 boolean 标记法`。 这种方法更容易理解,在面试中更容易写出来。 ### 迭代法中序遍历 @@ -311,60 +311,59 @@ class Solution: return result ``` -> 中序遍历,统一迭代(Set标记法): +> 中序遍历,统一迭代(boolean 标记法): ```python class Solution: def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]: values = [] - stack = [] if root is None else [root] - popped_nodes = set() # 用于记录一个节点是否被 pop() 过 + stack = [(root, False)] if root else [] # 多加一个参数,False 为默认值,含义见下文 while stack: - node = stack.pop() - # 说明节点是之前被pop过又被加回来,现在又要出栈,就可以直接收割了, - # 因为节点的左右儿子已经按次序入栈,节点的使命已经完成。 - if node in popped_nodes: + node, visited = stack.pop() # 多加一个 visited 参数,使“迭代统一写法”成为一件简单的事 + + if visited: # visited 为 True,表示该节点和两个儿子的位次之前已经安排过了,现在可以收割节点了 values.append(node.val) continue - popped_nodes.add(node) # 记录第一次出栈,第一次出栈的目的是为了把左右儿子和自己按次序入栈 + # visited 当前为 False, 表示初次访问本节点,此次访问的目的是“把自己和两个儿子在栈中安排好位次”。 + # 中序遍历是'左中右',右儿子最先入栈,最后出栈。 + if node.right: + stack.append((node.right, False)) - if node.right: # 中序遍历是'左中右',右儿子最先入栈,最后出栈 - stack.append(node.right) - - stack.append(node) # 把自己加回到栈中,位置居中 + stack.append((node, True)) # 把自己加回到栈中,位置居中。同时,设置 visited 为 True,表示下次再访问本节点时,允许收割 if node.left: - stack.append(node.left) # 左儿子最后入栈,最先出栈 + stack.append((node.left, False)) # 左儿子最后入栈,最先出栈 return values ``` -> 后序遍历,统一迭代(Set标记法): +> 后序遍历,统一迭代(boolean 标记法): ```python class Solution: def postorderTraversal(self, root: Optional[TreeNode]) -> List[int]: values = [] - stack = [] if root is None else [root] - popped_nodes = set() # 用于记录一个节点是否被 pop() 过 + stack = [(root, False)] if root else [] # 多加一个参数,False 为默认值,含义见下文 while stack: - node = stack.pop() - # 说明节点是之前被pop过又被加回来,现在又要出栈,就可以直接收割了, - # 因为节点的左右儿子已经按次序入栈,节点的使命已经完成。 - if node in popped_nodes: + node, visited = stack.pop() # 多加一个 visited 参数,使“迭代统一写法”成为一件简单的事 + + if visited: # visited 为 True,表示该节点和两个儿子位次之前已经安排过了,现在可以收割节点了 values.append(node.val) continue - popped_nodes.add(node) # 记录第一次出栈,第一次出栈的目的是为了把左右儿子和自己按次序入栈 - - stack.append(node) # 后序遍历是'左右中',节点自己最先入栈,最后出栈 + # visited 当前为 False, 表示初次访问本节点,此次访问的目的是“把自己和两个儿子在栈中安排好位次” + # 后序遍历是'左右中',节点自己最先入栈,最后出栈。 + # 同时,设置 visited 为 True,表示下次再访问本节点时,允许收割。 + stack.append((node, True)) if node.right: - stack.append(node.right) # 右儿子位置居中 + stack.append((node.right, False)) # 右儿子位置居中 if node.left: - stack.append(node.left) # 左儿子最后入栈,最先出栈 + stack.append((node.left, False)) # 左儿子最后入栈,最先出栈 + + return values ``` ### Go: