leetcode198:打家劫舍

思想:

判断nums的个数,若等于0直接返回,若等于1直接返回,若等于2取其中最大一个返回。

1.当nums中元素大于2,定义变量dp存放每次结果的最大值。原始dp[0]=nums[0],dp[1]=max(nums[0],nums[1])。为何dp[1]不等于nums[1]呢?当nums[0]+nums[3]>nums[1]+nums[3]时,我们应该要的是nums[0]+nums[3]。所以dp[1]存放的是nums[0]和num[1]中大的那一个。

2.当进入for循环时,同样也需比较。举一个列子,若nums[0]+nums[2]<nums[1],此时应要nums[1]。然后再继续比较,每次存下来的是最大的值。

class Solution:
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums) == 0:
            return 0
        if len(nums) <= 2:
            return max(nums)
        dp = [0 for i in range(0, 2)]
        dp[0] = nums[0]
        dp[1] = max(nums[1], nums[0])
        for i in range(2, len(nums)):
            dp[i % 2] = max(dp[(i - 1) % 2], dp[(i - 2) % 2] + nums[i])
        return dp[(len(nums) - 1) % 2]

大佬们的脑袋瓜是真的很奇妙。。。我还是比不了的呜呜呜呜呜

猜你喜欢

转载自blog.csdn.net/weixin_43160613/article/details/83684519