1. 題目描述

注: 屬於動態規劃, 儲存已解決的子問題的答案,在需要時再找出已求得的答案, 這樣就可以避免大量的重複計算, 節省時間.

可以用一個表來記錄所有已解的子問題的答案, 不管該子問題以後是否被用到, 只要它被計算過, 就將其結果填入表中.

2. 程式碼

 1 class Solution:
 2     def rob(self, nums: List[int]) -> int:
 3         n = len(nums)
 4         if n == 0:
 5             return n
 6         elif n == 1:
 
 7             return nums[0]
 8         elif n == 2:
 9             return max(nums[0], nums[1])
10         
11         dp = [0] * (n + 1)
12         dp[0] = 0
13         dp[1] = nums[0]
14         for i in range(2, n+1):
15             dp[i] = max(dp[i-2]+nums[i-1], dp[i-1])
16         return dp[n]

思路: dp表示當前所能取得的最大錢數.

重點是15行, 取得是 當前元素nums[i-1]和上上個房間的最大值dp[i-2]的和 跟 上個房間的最大值dp[i-1] 的最大值.