mirror of
https://github.com/youngyangyang04/leetcode-master.git
synced 2025-07-08 16:54:50 +08:00
动态规划章的 0337.打家劫舍3 的 Python 版本代码出现小错误,更新了我的代码
This commit is contained in:
@ -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
|
||||
|
Reference in New Issue
Block a user