DP-LeetCode337. 打家劫舍 III

1、题目描述

https://leetcode-cn.com/problems/house-robber-iii/

在上次打劫完一条街道之后和一圈房屋后，小偷又发现了一个新的可行窃的地区。这个地区只有一个入口，我们称之为“根”。 除了“根”之外，每栋房子有且只有一个“父“房子与之相连。一番侦察之后，聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫，房屋将自动报警。

计算在不触动警报的情况下，小偷一晚能够盗取的最高金额。

2、代码详解

1. 由于房屋是树状的，因此，我们可以使用遍历树的传统方法进行遍历(前序、中序、后续)
2. 简单的思路是，从树低进行往上遍历，拿到最优的打劫值。可以选用后续遍历
3. 得到每一节点的最优值，最后选取最优的结果

1.dp[i]代表该节点及以下拿到的最多的钱

2.动态方程：

• 2.1 dp[0]代表不偷该节点拿到最多的钱，则儿子节点偷不偷都ok。dp[0] = max(left[0], left[1]) + max(right[0], right[1])
• 2.2 dp[1]代表偷了该节点拿到最多的钱，则儿子节点都不能被偷。dp[1] = value + left[0] + right[0] (value代表该节点的价值)
• left[0]代表不偷取左孩子拿到最高的金额  ​ left[1]代表偷取左孩子拿到最高的金额  ​
• right[0]代表不偷取右孩子拿到最高的金额  ​ right[1]代表偷取右孩子拿到最高的金额

3.初始化：当前树的形状为空的时候，直接返回dp[0, 0]

# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution(object):
    def rob(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        def postTrasval(root):  # 从树低进行往上遍历，拿到最优的打劫值。可以选用后续遍历
            dp = [0, 0]  # dp[i]代表该节点及以下拿到的最多的钱
            if not root:
                return dp
            left = postTrasval(root.left)
            right = postTrasval(root.right)

            # dp[0]代表不偷该节点拿到最多的钱，则儿子节点偷不偷都ok。
            dp[0] = max(left[0], left[1]) + max(right[0], right[1])
            # dp[1]代表偷了该节点拿到最多的钱，则儿子节点都不能被偷。
            dp[1] = root.val + left[0] + right[0]

            return dp

        dp = postTrasval(root)
        return max(dp[0], dp[1])

if __name__ == '__main__':

    T = TreeNode(3)
    T.left = TreeNode(2)
    T.right = TreeNode(3)
    T.left.right = TreeNode(3)
    T.right.right = TreeNode(1)


    s = Solution()
    print(s.rob(T))

https://zhuanlan.zhihu.com/p/130743652