题目:213. 打家劫舍 II
链接:https://leetcode-cn.com/problems/house-robber-ii
你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。 给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。 示例 1: 输入: [2,3,2] 输出: 3 解释: 你不能先偷窃 1 号房屋(金额 = 2),然后偷窃 3 号房屋(金额 = 2), 因为他们是相邻的。
解题:
1、dp问题。分为两种情况:偷第一家(不能偷最后一家)和不偷第一家(可以偷最后一家)。
dp公式和昨天一样:dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
代码:
class Solution:
def rob(self, nums: List[int]) -> int:
if len(nums) == 0:
return 0
res = 0
# 偷第一家
pre, cur = 0, nums[0]
for i in range(1, len(nums) - 1):
pre, cur = cur, max(cur, pre + nums[i])
res = max(pre, cur)
# 不偷第一家
pre, cur = 0, 0
for i in range(1, len(nums)):
pre, cur = cur, max(cur, pre + nums[i])
res = max(res, pre, cur)
return res