二叉树的统一迭代法.md 使用"boolean标记法"并调整相关Python代码实现

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：