198. 打家劫舍

Easy

思路

先审题,一般题目问最大的金额,最多的路径,解的个数之类的问题,很多情况是考察动态规划。抛开已有认知,还是先分析题目。

这样的话,暴破得出所有解法, 然后求出和,再进行比较是一种方法,但是应该会TLE,过!需改善思路。

分解问题,以[2,7,9,3,1] 为例,从前到后一个房子一个房子抢,分析抢到当前房子能够抢到的最大,只考虑到当前,不考虑后面的

i:index, h:代表房子, 箭头:当前偷到的位置, m:当前位置能偷到的最多的钱


第1步
i    0  1  2  3  4
h    2  7  9  3  1 
      
m    2
假如只有1个房子那肯定抢


第2步
i    0  1  2  3  4
h    2  7  9  3  1 
         
m    2  7
到第二个房子怎么考虑抢或不抢
假如拿了h0(第一个房间)就不能拿h1应为不能拿相邻的
那h1比h0值钱我肯定就放弃h0抢h1所以到第二个房间的最多是7


第3步
i    0  1  2  3  4
h    2  7  9  3  1 
            
m    2  7  11
到第三个房子了那h2抢不抢?
1. : h2抢那h1不能抢取h0位置的能抢的最大和自身位置和m2 = h2 + m0 = 11
2. 不抢:  那就抢h1为7
两种情况取大值故11


第4步
i    0  1  2  3  4
h    2  7  9  3  1 
               
m    2  7  11 11 12
后面同理了简单过一遍
1. : m3 = h3 + m1 = 3 + 7 = 10
2. 不抢:  11
max(10,11)  11


第5步
i    0  1  2  3  4
h    2  7  9  3  1 
               
m    2  7  11 11 12
1. : m4 = h4 + m2 = 1 + 11 = 12
2. 不抢:  11
max(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]