[LeetCode]198.打家劫舍(python)

最新推荐文章于 2024-09-06 09:22:51 发布

xyhaaab

最新推荐文章于 2024-09-06 09:22:51 发布

阅读量109

点赞数 1

文章标签： leetcode python 算法

本文链接：https://blog.csdn.net/xyhaaab/article/details/141189647

版权

1.代码

class Solution:
    def rob(self, nums: List[int]) -> int:
        length=len(nums)
        if(length==0):
            return 0
        if(length==1):
            return nums[0]
        if(length==2):
            return max(nums[0],nums[1])
        result = []
        result.append(0)
        result.append(nums[0])
        result.append(max(nums[0],nums[1]))
        for i in range(3,length+1):
            result.append(max(result[i-1],result[i-2]+nums[i-1]))
        return result[length]

2.思路

对长度为0，1，2的情况单独讨论，当长度>=3，分为抢劫最后一家和不抢劫最后一家两种情况，若抢劫最后一家则无法抢劫倒数第二家。