198. House Robber
Title
你是一個專業(yè)的小偷,計劃偷竊沿街的房屋。每間房內(nèi)都藏有一定的現(xiàn)金,影響你偷竊的唯一制約因素就是相鄰的房屋裝有相互連通的防盜系統(tǒng),如果兩間相鄰的房屋在同一晚上被小偷闖入,系統(tǒng)會自動報警。
給定一個代表每個房屋存放金額的非負整數(shù)數(shù)組,計算你不觸動警報裝置的情況下 ,一夜之內(nèi)能夠偷竊到的最高金額。
示例 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 。
Solve
動態(tài)規(guī)劃+滾動數(shù)組:
還是先從最簡單的情況開始考慮:
用dp[i]表示前i間房子能偷竊到的最高總金額,有如下狀態(tài)轉(zhuǎn)移方程:
dp[i]=max(dp[i?2]+nums[i],dp[i?1])dp[i]=max(dp[i?2]+nums[i],dp[i?1]) dp[i]=max(dp[i?2]+nums[i],dp[i?1])
邊界條件為:dp[0]=nums[0],只有一間房屋則偷竊該房屋dp[0]=nums[0],只有一間房屋則偷竊該房屋dp[0]=nums[0],只有一間房屋則偷竊該房屋dp[1]=max(nums[0],nums[1]),兩間房屋選擇其中金額較高的偷竊dp[1]=max(nums[0],nums[1]),兩間房屋選擇其中金額較高的偷竊dp[1]=max(nums[0],nums[1]),兩間房屋選擇其中金額較高的偷竊
最終的答案即為 dp[n?1],其中 n 是數(shù)組的長度。
考慮到每間房屋的最高總金額只和該房屋的前兩間房屋的最高總金額相關,因此可以使用滾動數(shù)組,在每個時刻只需要存儲前兩間房屋的最高總金額。
復雜度分析
時間復雜度:O(n),其中 n 是數(shù)組長度。只需要對數(shù)組遍歷一次。
空間復雜度:O(1)。使用滾動數(shù)組,可以只存儲前兩間房屋的最高總金額,而不需要存儲整個數(shù)組的結(jié)果,因此空間復雜度是 O(1)。
Code
class Solution:def rob(self, nums: List[int]) -> int:if not nums:return 0length = len(nums)if length == 1:return nums[0]first, second = nums[0], max(nums[0], nums[1])for i in range(2, length):first, second = second, max(first + nums[i], second)return second總結(jié)
以上是生活随笔為你收集整理的198. House Robber的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 394. Decode String
- 下一篇: 记录第一次面试