diff --git a/problems/0337.打家劫舍III.md b/problems/0337.打家劫舍III.md
index d2add232..fff8d3da 100644
--- a/problems/0337.打家劫舍III.md
+++ b/problems/0337.打家劫舍III.md
@@ -353,18 +353,30 @@ class Solution:
 #         self.left = left
 #         self.right = right
 class Solution:
-    def rob(self, root: TreeNode) -> int:
-        result = self.rob_tree(root)
-        return max(result[0], result[1])
-    
-    def rob_tree(self, node):
-        if node is None:
-            return (0, 0) # (偷当前节点金额，不偷当前节点金额)
-        left = self.rob_tree(node.left)
-        right = self.rob_tree(node.right)
-        val1 = node.val + left[1] + right[1] # 偷当前节点，不能偷子节点
-        val2 = max(left[0], left[1]) + max(right[0], right[1]) # 不偷当前节点，可偷可不偷子节点
-        return (val1, val2)
+    def rob(self, root: Optional[TreeNode]) -> int:
+        # dp数组（dp table）以及下标的含义：
+        # 1. 下标为 0 记录 **不偷该节点** 所得到的的最大金钱
+        # 2. 下标为 1 记录 **偷该节点** 所得到的的最大金钱
+        dp = self.traversal(root)
+        return max(dp)
+
+    # 要用后序遍历, 因为要通过递归函数的返回值来做下一步计算
+    def traversal(self, node):
+        
+        # 递归终止条件，就是遇到了空节点，那肯定是不偷的
+        if not node:
+            return (0, 0)
+
+        left = self.traversal(node.left)
+        right = self.traversal(node.right)
+
+        # 不偷当前节点, 偷子节点
+        val_0 = max(left[0], left[1]) + max(right[0], right[1])
+
+        # 偷当前节点, 不偷子节点
+        val_1 = node.val + left[0] + right[0]
+
+        return (val_0, val_1)
 ```
 
 ### Go