213. 打家劫舍 II
和第192打家劫舍类似,当房间数为1时不用选择,房间数为2时选较大的一家。
在大于2家时由于首尾相连,可以分成两种情况,选中第一家,不选第一家,
选中第一家:
则一定不能选最后一家,令nums1 = nums[:-1]
不选第一家:
nums2 = nums[1:]
从两种情况中选择较大的
class Solution:
def rob(self, nums: List[int]) -> int:
n = len(nums) - 1
if n <= 0:
return nums[0]
elif n == 1:
return max(nums[0], nums[1])
nums1 = nums[1:]
nums2 = nums[:-1]
dp1 = [0] * n
dp1[0] = nums1[0]
dp1[1] = max(nums1[0], nums1[1])
for i in range(2, n):
dp1[i] = max(dp1[i - 2] + nums1[i], dp1[i - 1])
dp2 = [0] * n
dp2[0] = nums2[0]
dp2[1] = max(nums2[0], nums2[1])
for i in range(2, n):
dp2[i] = max(dp2[i - 2] + nums2[i], dp2[i - 1])
return max(dp1[-1], dp2[-1])