题意:如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。例如: [1,2,3,1],偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。最高金额 = 1 + 3 = 4 。
方法一:动态规划法
先找状态方程,由于相邻的房屋不能一同被偷,所以第n号房屋偷了,n-1房屋不能偷。所以偷到第n号房间时的金额为dp[n]=max(dp[n-1],dp[n-2]+nums[n-1])(注意是nums[n-1])
class Solution:
def rob(self, nums: List[int]) -> int:
if len(nums)==0:
return 0 #没有数据,则返回0
dp = [0]*(len(nums)+1) (因为dp[1]=nums[0],所以在原长度加上了dp[0]=0)
dp[1]=nums[0]
for i in range(2,len(nums)+1):
dp[i]=max(dp[i-1],dp[i-2]+nums[i-1])
return dp[len(nums)]
只要理解了dp[1]=nums[0]就不会搞错长度了。
dp[len(nums)]=nums(len(nums)-1)即最后一个数据
时间复杂度和空间复杂度为O(N)
方法二:优化空间O(1)
class Solution:
def rob(self, nums: List[int]) -> int:
cur,pre=0,0
for i in nums:
cur,pre=max(cur,pre+i),cur return cur
发现dp[n]只和dp[n-1]和dp[n-2]有关系,所以可以假设cur和pre两个变量来交替记录,将空间复杂度降到O(1),即cur和pre使用常数大小的额外空间,时间复杂度为O(N),遍历nums需要线性时间。