题解 | #打家劫舍(二)#

动态规划,dp[i] = max(dp[i + 2] + nums[i], dp[i + 1]),由于围城了一个圈,第一家和最后一家算作相邻,即求 nums[1:] 和 nums[:-1] 上的最大 dp 值

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param nums int整型一维数组 
# @return int整型
#
class Solution:
    def rob(self , nums: List[int]) -> int:
        # write code here
        if len(nums) == 1:
            return nums[0]
        return max(self.sub_rob(nums[:-1]), self.sub_rob(nums[1:]))
    
    def sub_rob(self, nums: List[int]) -> int:
        dp = [0 for _ in range(len(nums) + 2)]
        for i in range(len(nums) - 1, -1, -1):
            dp[i] = max(dp[i + 2] + nums[i], dp[i + 1])
        return dp[0]