mirror of
https://github.com/devenperez/leetcode.git
synced 2026-06-13 14:57:08 +00:00
Deven
24 lines
548 B
Python
24 lines
548 B
Python
class Solution(object):
|
|
def rob(self, nums):
|
|
"""
|
|
:type nums: List[int]
|
|
:rtype: int
|
|
"""
|
|
|
|
if len(nums) < 3:
|
|
return max(nums)
|
|
|
|
# Subproblem: only nums 0 to i exist and i is always robbed
|
|
dp = [0] * 3 # Optimized to only hold the last 3 entries
|
|
|
|
dp[0] = nums[0]
|
|
dp[1] = nums[1]
|
|
dp[2] = nums[2] + nums[0]
|
|
|
|
|
|
for i in range(3, len(nums)):
|
|
x = i % 3
|
|
dp[x] = nums[i] + max(dp[(x - 2) % 3], dp[x])
|
|
|
|
return max(dp)
|
|
|