二叉树的统一迭代法.md 加入"Set标记法"和相关Python代码实现

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：
 
 > 前序遍历统一迭代法