思想:
判断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]
大佬们的脑袋瓜是真的很奇妙。。。我还是比不了的呜呜呜呜呜