LeetCode213_打家劫舍 II

1. 题目

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警 。

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

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

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

示例 3：
输入：nums = [1,2,3]
输出：3
 
提示：
1 <= nums.length <= 100
0 <= nums[i] <= 1000

2. 题解

2.1 解法1： 动态规划

from typing import List


class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)
        if n == 0:
            return 0
        if n <= 2:
            return max(nums)
        return max(self.get_max_money(nums[1:]), self.get_max_money(nums[:-1]))

    def get_max_money(self, nums: List[int]) -> int:
        dp = [0] * len(nums)
        dp[0] = nums[0]
        dp[1] = max(nums[0], nums[1])
        for i in range(2, len(nums)):
            dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
        return dp[-1]


if __name__ == "__main__":
    s = Solution()
    a = s.rob([1])
    print(a)

2.2 解法2

from typing import List


class Solution:
    def rob(self, nums: List[int]) -> int:
        count = len(nums)
        if count == 0:
            return 0
        if count == 1:
            return nums[0]

        dp1 = self.helper(nums[:-1])
        dp2 = self.helper(nums[1:])
        return max(dp1, dp2)

    def helper(self, nums: List[int]) -> int:
        count = len(nums)
        if count == 0:
            return 0
        dp = [0] * (count + 1)
        dp[-2] = nums[-1]
        for i in range(count - 2, -1, -1):
            dp[i] = max(dp[i + 1], dp[i + 2] + nums[i])
        return dp[0]


if __name__ == "__main__":
    s = Solution()
    a = s.rob([1])
    print(a)