Leetcode 213.打家劫舍 II（House Robber II）

Leetcode 213.打家劫舍 II

1 题目描述（Leetcode题目链接）

  你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。给定一个代表每个房屋存放金额的非负整数数组，计算你在不触动警报装置的情况下，能够偷窃到的最高金额。

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

2 题解

  相对于打家劫舍I，本题增加了首尾两个房子不能同时偷的限制，那么我们可以分情况讨论，对于第一间房子，要么偷，要么不偷，那就变成了两个打家劫舍I的问题，按照之前的状态转移方程做即可：
$$DP[i]=max(DP[i-1],DP[i-2]+nums[i])$$

class Solution:
    def rob(self, nums: List[int]) -> int:
        length = len(nums)
        if length == 0: return 0
        if length == 1: return nums[0]
        DP1 = [0] * (length + 1)
        DP2 = [0] * (length + 1)
        for i in range(2, length + 1):
            DP1[i] = max(DP1[i - 1], DP1[i - 2] + nums[i - 2])
            DP2[i] = max(DP2[i - 1], DP2[i - 2] + nums[i - 1])
        return max(DP1[length], DP2[length])