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:])
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容