题意:一组直线排列的房屋,一个小偷要进屋偷钱,如果小偷偷了相邻两座房屋就会触发报警系统,问在不触发报警系统的前提下小偷最多可以偷到多少钱。
难度:esay
思路:这是一道标准的动态规划问题,创建一个list保存小偷到每个房间能拿到最多的钱,第一个房间为本身,第二个为前两个房间较大者。第i个房间便为(状态转移方程):
dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
class Solution(object):
def rob(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
dp = [0] * len(nums)
if not nums:
return 0
elif len(nums) == 1:
return nums[0]
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]