leetcode -198. 打家劫舍

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。

首先 ,思路是:
result = max(nums[index] +solve(index-1, nums), solve(index-2, nums))
这样写会重复计算,效率极低
所以用列表存储每一次计算的solve的值。

class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)
        
        if n == 0:
            return 0
        elif n == 1:
            return nums[0]
        
        dp = [0] * n
        dp[0] = nums[0]
        dp[1] = max(nums[:2])
        
        for i in range(2, n):
            # dp[i] = max(dp[: i - 1]) + nums[i]
            dp[i] = max(dp[i - 2], dp[i - 3]) + nums[i]
        
        return max(dp[-2:])

你可能感兴趣的:(leetcode -198. 打家劫舍)