198. 打家劫舍
Easy
思路
先审题,一般题目问最大的金额,最多的路径,解的个数之类的问题,很多情况是考察动态规划。抛开已有认知,还是先分析题目。
- 第一次选择,可以选任意
- 第二次选择,不能选择相邻位置
这样的话,暴破得出所有解法, 然后求出和,再进行比较是一种方法,但是应该会TLE,过!需改善思路。
分解问题,以[2,7,9,3,1] 为例,从前到后一个房子一个房子抢,分析抢到当前房子能够抢到的最大,只考虑到当前,不考虑后面的
i:index, h:代表房子, 箭头:当前偷到的位置, m:当前位置能偷到的最多的钱第1步i 0 1 2 3 4h 2 7 9 3 1↑m 2假如只有1个房子那肯定抢第2步i 0 1 2 3 4h 2 7 9 3 1↑m 2 7到第二个房子,怎么考虑抢或不抢?假如拿了h0(第一个房间)就不能拿h1,应为不能拿相邻的;那h1比h0值钱,我肯定就放弃h0,抢h1。所以到第二个房间的最多是7第3步i 0 1 2 3 4h 2 7 9 3 1↑m 2 7 11到第三个房子了,那h2抢不抢?1. 抢: h2抢,那h1不能抢,取h0位置的能抢的最大和自身位置和,m2 = h2 + m0 = 112. 不抢: 那就抢h1,为7两种情况取大值,故11第4步i 0 1 2 3 4h 2 7 9 3 1↑m 2 7 11 11 12后面同理了,简单过一遍1. 抢: m3 = h3 + m1 = 3 + 7 = 102. 不抢: 11max(10,11) 故 11第5步i 0 1 2 3 4h 2 7 9 3 1↑m 2 7 11 11 121. 抢: m4 = h4 + m2 = 1 + 11 = 122. 不抢: 11max(12,11) 故 12
那这道题的最终答案就是m[-1],而且我们通过遍历h列表计算出m列表,
通项公式:
$ i = 0: m[0] = h[0] $
$ i = 1: m[1] = max(h[0], h[1]) $
$ i > 1: m[i] = max(h[i]+m[i-2], m[i-1]) $
代码
python3
class Solution:
def rob(self, nums: List[int]) -> int:
if len(nums) == 0:
return 0
if len(nums) == 1:
return nums[0]
dp = [0] * len(nums)
# base case
dp[0] = nums[0]
dp[1] = max(nums[0], nums[1])
for i in range(2, len(nums)):
dp[i] = max(nums[i] + dp[i-2], dp[i-1])
return dp[-1]
