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))
相关推荐
不見星空14 分钟前
leetcode 每日一题 1865. 找出和为指定值的下标对
算法·leetcode
我爱Jack25 分钟前
时间与空间复杂度详解:算法效率的度量衡
java·开发语言·算法
DoraBigHead2 小时前
小哆啦解题记——映射的背叛
算法
Heartoxx2 小时前
c语言-指针与一维数组
c语言·开发语言·算法
孤狼warrior3 小时前
灰色预测模型
人工智能·python·算法·数学建模
京东云开发者3 小时前
京东零售基于国产芯片的AI引擎技术
算法
chao_7894 小时前
回溯题解——子集【LeetCode】二进制枚举法
开发语言·数据结构·python·算法·leetcode
十盒半价4 小时前
从递归到动态规划:手把手教你玩转算法三剑客
javascript·算法·trae
GEEK零零七4 小时前
Leetcode 1070. 产品销售分析 III
sql·算法·leetcode
凌肖战4 小时前
力扣网编程274题:H指数之普通解法(中等)
算法·leetcode