LeetCode刷题日记（213. 打家劫舍 II）

'''
你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。
这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。
同时，相邻的房屋装有相互连通的防盗系统，
如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组，
计算你在不触动警报装置的情况下，今晚能够偷窃到的最高金额。
'''
from typing import List
class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)
        if n == 0:
            return 0
        if n == 1:
            return nums[0]

        def rob0(nums: List[int]) -> int:
            n = len(nums)
            if n == 0:
                return 0
            if n == 1:
                return nums[0]
            # dp[i]表示前i个房子能偷到的最大金额
            dp = [0 for i in range(n)]
            # 第 0,1 个房子只能偷一个
            dp[0] = nums[0]
            dp[1] = max(nums[0], nums[1])
            for i in range(2, n):
                # 不偷第i个房子， dp[i]=dp[i-1]
                # 偷第i个房子， dp[i]=dp[i-2]+nums[i]
                dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
            return dp[-1]
        return max(rob0(nums[1:]), rob0(nums[:-1]))

对198.打家劫舍的改编。分别计算除了第一个和除了最后一个的两种情况，返回两者最大值。