From cded6c5c803e5d6de7a47e020e94ce03fcc68432 Mon Sep 17 00:00:00 2001
From: ExplosiveBattery <641370196@qq.com>
Date: Thu, 9 Jun 2022 00:58:38 +0800
Subject: [PATCH] =?UTF-8?q?Update=200110.=E5=B9=B3=E8=A1=A1=E4=BA=8C?=
 =?UTF-8?q?=E5=8F=89=E6=A0=91.md=20=20python=20code=20via=20iterate?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

In the original method, we need to traversal every node and write the function named getDepth to get the depth of all sub trees in traverse method too.
But there is more suitable uniform iteration traversal algorithm, I use the map struct in the code segment where the node is Null.
If you have problem in understand, please feel free to communicate with me.
---
 problems/0110.平衡二叉树.md | 46 +++++++++++---------------------
 1 file changed, 16 insertions(+), 30 deletions(-)

diff --git a/problems/0110.平衡二叉树.md b/problems/0110.平衡二叉树.md
index d98ff8a9..1b997643 100644
--- a/problems/0110.平衡二叉树.md
+++ b/problems/0110.平衡二叉树.md
@@ -531,40 +531,26 @@ class Solution:
 迭代法：
 ```python
 class Solution:
-    def isBalanced(self, root: TreeNode) -> bool:
-        st = []
+    def isBalanced(self, root: Optional[TreeNode]) -> bool:
         if not root:
             return True
-        st.append(root)
-        while st:
-            node = st.pop() #中
-            if abs(self.getDepth(node.left) - self.getDepth(node.right)) > 1:
-                return False
-            if node.right:
-                st.append(node.right) #右（空节点不入栈）
-            if node.left:
-                st.append(node.left) #左（空节点不入栈）
-        return True
-    
-    def getDepth(self, cur):
-        st = []
-        if cur:
-            st.append(cur)
-        depth = 0
-        result = 0
-        while st:
-            node = st.pop()
+
+        height_map = {}
+        stack = [root]
+        while stack:
+            node = stack.pop()
             if node:
-                st.append(node) #中
-                st.append(None)
-                depth += 1
-                if node.right: st.append(node.right) #右
-                if node.left: st.append(node.left) #左
+                stack.append(node)
+                stack.append(None)
+                if node.left: stack.append(node.left)
+                if node.right: stack.append(node.right)
             else:
-                node = st.pop()
-                depth -= 1
-            result = max(result, depth)
-        return result
+                real_node = stack.pop()
+                left, right = height_map.get(real_node.left, 0), height_map.get(real_node.right, 0)
+                if abs(left - right) > 1:
+                    return False
+                height_map[real_node] = 1 + max(left, right)
+        return True
 ```