leetcode学习笔记(打家劫舍 II)

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])