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))
相关推荐
JoernLee5 分钟前
机器学习算法:支持向量机SVM
人工智能·算法·机器学习
V我五十买鸡腿14 分钟前
顺序栈和链式栈
c语言·数据结构·笔记·算法
我爱一条柴ya34 分钟前
【AI大模型】线性回归:经典算法的深度解析与实战指南
人工智能·python·算法·ai·ai编程
三维重建-光栅投影2 小时前
VS中将cuda项目编译为DLL并调用
算法
课堂剪切板4 小时前
ch03 部分题目思路
算法
山登绝顶我为峰 3(^v^)35 小时前
如何录制带备注的演示文稿(LaTex Beamer + Pympress)
c++·线性代数·算法·计算机·密码学·音视频·latex
Two_brushes.6 小时前
【算法】宽度优先遍历BFS
算法·leetcode·哈希算法·宽度优先
森焱森9 小时前
水下航行器外形分类详解
c语言·单片机·算法·架构·无人机
QuantumStack10 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法
写个博客11 小时前
暑假算法日记第一天
算法