动态规划章的 0337.打家劫舍3 的 Python 版本代码出现小错误，更新了我的代码

problems/0337.打家劫舍III.md

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