Leetcode 337. 打家劫舍 III

心路历程:

还是经典的动态规划题,建模思路:

状态:第i个房间;j偷或者不偷

候选动作:偷 或者不偷

返回值:当前状态获得的最大金额

注意的点:

1、当计算当前结点被偷时,注意node.val不要加重复了

2、objective对象也是hashable的,可以@cache处理

解法:动态规划

python 复制代码
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def rob(self, root: Optional[TreeNode]) -> int:

        @cache
        def dp(node, j):
            # if not node: return 0
            if j == 0: 
                res1, res2 = 0, 0
                if node.left: res1 = max(dp(node.left, 1), dp(node.left, 0))
                if node.right: res2 = max(dp(node.right, 1), dp(node.right, 0))
                return res1 + res2
            else:  # 这块的node.val不要加重复了!!!
                res1, res2 = 0, 0
                if node.left: res1 = dp(node.left, 0)
                if node.right: res2 = dp(node.right, 0)
                return res1 + res2 + node.val
        return max(dp(root, 1), dp(root, 0))
相关推荐
算AI18 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
hyshhhh20 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
杉之21 小时前
选择排序笔记
java·算法·排序算法
烂蜻蜓21 小时前
C 语言中的递归:概念、应用与实例解析
c语言·数据结构·算法
OYangxf21 小时前
图论----拓扑排序
算法·图论
我要昵称干什么21 小时前
基于S函数的simulink仿真
人工智能·算法
AndrewHZ1 天前
【图像处理基石】什么是tone mapping?
图像处理·人工智能·算法·计算机视觉·hdr
念九_ysl1 天前
基数排序算法解析与TypeScript实现
前端·算法·typescript·排序算法
守正出琦1 天前
日期类的实现
数据结构·c++·算法
ChoSeitaku1 天前
NO.63十六届蓝桥杯备战|基础算法-⼆分答案|木材加工|砍树|跳石头(C++)
c++·算法·蓝桥杯