一、题目描述
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
示例 1
输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
示例 2
输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
偷窃到的最高金额 = 2 + 9 + 1 = 12 。
提示:
1 <= nums.length <= 100
0 <= nums[i] <= 400
二、代码
代码如下:
class Solution:
def rob(self, nums: List[int]) -> int:
length = len(nums)
for i in range(2,length):
nums[i] = max(nums[0:i-1]) + nums[i]
print(nums[length-1])
print(nums[length-2])
print(max(nums[length-1],nums[length-2]))
return max(nums[length-1],nums[length-2])
三、解题思路
本题需要用到动态规划的思想,如果我们使用多重循环遍历的方法得出每一种组合的情况,那么则会导致超出时间;所以则需要换一种处理方式,我们发现,如果当前的房屋i( i >= 2)
要取得最大值,则需要加上 0 ~ i-1
中的最大值,所以我们的解题思路可以是找出当前房屋i对应的最大值。
我们从房屋2
开始,那么房屋2
的最大值只可能有1种情况,最大值 = 房屋2
+ 房屋0
房屋3
的最大值则可能有多种情况 :
最大值 = 房屋3
+ 房屋1
最大值 = 房屋3
+ 房屋0
但是可以将其合并为:
最大值 = 房屋3
+ 房屋[0 , 2)
的最大值
以此类推,房屋n
的最大值为:
最大值 = 房屋n
+ 房屋[0 , n-1)
的最大值
这是一个从前往后的过程,我们在计算出前面房屋的最大值后,可以用于计算后面房屋的最大值,所以此时可以只用一次for循环就能得出最后答案,最后的结果会被累加到数组末尾。