diff --git a/problems/二叉树的统一迭代法.md b/problems/二叉树的统一迭代法.md index 13c50737..a1ac9dd2 100644 --- a/problems/二叉树的统一迭代法.md +++ b/problems/二叉树的统一迭代法.md @@ -27,7 +27,12 @@ **那我们就将访问的节点放入栈中,把要处理的节点也放入栈中但是要做标记。** -如何标记呢,**就是要处理的节点放入栈之后,紧接着放入一个空指针作为标记。** 这种方法也可以叫做标记法。 +如何标记呢? + +* 方法一:**就是要处理的节点放入栈之后,紧接着放入一个空指针作为标记。** 这种方法可以叫做`空指针标记法`。 + +* 方法二:**当一个节点被`pop()`后,把该节点放入一个`Set`中,表示该节点被处理过了,下次再处理这个节点时,直接收割。** +这种方法可以叫做`Set标记法`,样例代码见下文`Python Set标记法`。 方法二更容易理解,在面试中更容易写出来。 ### 迭代法中序遍历 @@ -234,7 +239,7 @@ class Solution { ### Python: -迭代法前序遍历: +> 迭代法前序遍历(空指针标记法): ```python class Solution: def preorderTraversal(self, root: TreeNode) -> List[int]: @@ -257,7 +262,7 @@ class Solution: return result ``` -迭代法中序遍历: +> 迭代法中序遍历(空指针标记法): ```python class Solution: def inorderTraversal(self, root: TreeNode) -> List[int]: @@ -282,7 +287,7 @@ class Solution: return result ``` -迭代法后序遍历: +> 迭代法后序遍历(空指针标记法): ```python class Solution: def postorderTraversal(self, root: TreeNode) -> List[int]: @@ -306,6 +311,62 @@ class Solution: return result ``` +> 中序遍历,统一迭代(Set标记法): +```python +class Solution: + def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]: + values = [] + stack = [] if root is None else [root] + popped_nodes = set() # 用于记录一个节点是否被 pop() 过 + + while stack: + node = stack.pop() + # 说明节点是之前被pop过又被加回来,现在又要出栈,就可以直接收割了, + # 因为节点的左右儿子已经按次序入栈,节点的使命已经完成。 + if node in popped_nodes: + values.append(node.val) + continue + + popped_nodes.add(node) # 记录第一次出栈,第一次出栈的目的是为了把左右儿子和自己按次序入栈 + + if node.right: # 中序遍历是'左中右',右儿子最先入栈,最后出栈 + stack.append(node.right) + + stack.append(node) # 把自己加回到栈中,位置居中 + + if node.left: + stack.append(node.left) # 左儿子最后入栈,最先出栈 + + return values +``` + +> 后序遍历,统一迭代(Set标记法): +```python +class Solution: + def postorderTraversal(self, root: Optional[TreeNode]) -> List[int]: + values = [] + stack = [] if root is None else [root] + popped_nodes = set() # 用于记录一个节点是否被 pop() 过 + + while stack: + node = stack.pop() + # 说明节点是之前被pop过又被加回来,现在又要出栈,就可以直接收割了, + # 因为节点的左右儿子已经按次序入栈,节点的使命已经完成。 + if node in popped_nodes: + values.append(node.val) + continue + + popped_nodes.add(node) # 记录第一次出栈,第一次出栈的目的是为了把左右儿子和自己按次序入栈 + + stack.append(node) # 后序遍历是'左右中',节点自己最先入栈,最后出栈 + + if node.right: + stack.append(node.right) # 右儿子位置居中 + + if node.left: + stack.append(node.left) # 左儿子最后入栈,最先出栈 +``` + ### Go: > 前序遍历统一迭代法