213.打家劫舍-ii.py

#
# @lc app=leetcode.cn id=213 lang=python3
#
# [213] 打家劫舍 II
#
# https://leetcode-cn.com/problems/house-robber-ii/description/
#
# algorithms
# Medium (41.39%)
# Likes:    631
# Dislikes: 0
# Total Accepted:    112.8K
# Total Submissions: 266.8K
# Testcase Example:  '[2,3,2]'
#
# 你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈
# ，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警 。
# 
# 给定一个代表每个房屋存放金额的非负整数数组，计算你 在不触动警报装置的情况下 ，能够偷窃到的最高金额。
# 
# 
# 
# 示例 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 = [0]
# 输出：0
# 
# 
# 
# 
# 提示：
# 
# 
# 1 
# 0 
# 
# 
#

# @lc code=start
class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)
        if n == 1:
            return nums[0]
        dpa = [0] * len(nums) #选择第一家
        dpb = [0] * len(nums) #不选择第一家
        dpa[0] = nums[0]
        dpa[1] = nums[0]
        dpb[0] = 0
        dpb[1] = nums[1]
        for i in range(2, n - 1):
            dpa[i] = max(dpa[i - 2] + nums[i], dpa[i - 1])
            dpb[i] = max(dpb[i - 2] + nums[i], dpb[i - 1])
        if n > 2:
            dpa[-1] = max(dpa[-3], dpa[-2])
            dpb[-1] = max(dpb[-3] + nums[-1], dpb[-2])
        return max(dpa[-1], dpb[-1])

# @lc code=end