LeetCode-打家劫舍-动态规划

1、打家劫舍-

你是一个专业的小偷，计划偷窃沿街的房屋。每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组，计算你在不触动警报装置的情况下，能够偷窃到的最高金额。

示例 1:

输入: [1,2,3,1]
输出: 4
解释: 偷窃 1 号房屋 (金额 = 1) ，然后偷窃 3 号房屋 (金额 = 3)。偷窃到的最高金额 = 1 + 3 = 4 。
示例 2:

输入: [2,7,9,3,1]
输出: 12
解释: 偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9)，接着偷窃 5 号房屋 (金额 = 1)。偷窃到的最高金额 = 2 + 9 + 1 = 12 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/house-robber/

解析：

f[i]表示0-i的最大金额，f[i] = max(f[i - k] + nums[i]，f[i])，k为间隔数量

def rob1(nums):
    if not nums:
        return 0
    n = len(nums)
    f = list(nums)
    f[0] = nums[0]
    for i in range(n):
        for k in range(2, i + 1):
            if i >= k and f[i] < f[i - k] + nums[i]:
                f[i] = f[i - k] + nums[i]
    return max(f)

2、打家劫舍 II

同上，但是有区别：这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。

示例 1:

输入: [2,3,2]
输出: 3
解释: 你不能先偷窃 1 号房屋（金额 = 2），然后偷窃 3 号房屋（金额 = 2）, 因为他们是相邻的。
示例 2:

输入: [1,2,3,1]
输出: 4
解释: 你可以先偷窃 1 号房屋（金额 = 1），然后偷窃 3 号房屋（金额 = 3）。偷窃到的最高金额 = 1 + 3 = 4 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/house-robber-ii/

解析：

        环状排列意味着第一个房子和最后一个房子中只能选择一个偷窃，因此可以把此环状排列房间问题约化为两个单排排列房间子问题：

        1、在不偷窃第一个房子的情况下（即 nums[1:]nums[1:]），最大金额是 p1

        2、在不偷窃最后一个房子的情况下（即 nums[:n-1]nums[:n−1]），最大金额是 p2

        综合偷窃最大金额： 为以上两种情况的较大值，即 max(p1,p2) 。

 def rob2(nums):
        if not nums:
            return 0

        n = len(nums)
        f = list(nums)

        for i in range(1, n):
            for k in range(2, i + 1):
                if i - k >= 1 and f[i] < f[i - k] + nums[i]:
                    f[i] = f[i - k] + nums[i]

        f2 = list(nums)
        for i in range(n-1):
            for k in range(2, i + 1):
                if i - k >= 0 and f2[i] < f2[i - k] + nums[i]:
                    f2[i] = f2[i - k] + nums[i]

        return max(f + f2)

3、打家劫舍 III

本题不是动态规划，树的DFS

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

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

示例 1:

输入: [3,2,3,null,3,null,1]

     3
    / \
   2   3
    \   \ 
     3   1

输出: 7 
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
示例 2:

输入: [3,4,5,1,3,null,1]

     3
    / \
   4   5
  / \   \ 
 1   3   1

输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/house-robber-iii/

def rob(root):
    # res1为不包括根节点的最大值，res2为包括根节点的最大值
    def dfs(root):
        if not root:
            return 0, 0

        left1, left2 = dfs(root.left)
        right1, right2 = dfs(root.right)
        res1 = max(left1, left2) + max(right1, right2)
        res2 = left1 + right1 + root.val
        return res1, res2

    return max(dfs(root))