描述
假设你是一个专业的窃贼,准备沿着一条街打劫房屋。每个房子都存放着特定金额的钱。你面临的唯一约束条件是:相邻的房子装着相互联系的防盗系统,且 当相邻的两个房子同一天被打劫时,该系统会自动报警。
给定一个非负整数列表,表示每个房子中存放的钱, 算一算,如果今晚去打劫,在不触动报警装置的情况下, 你最多可以得到多少钱 。
您在真实的面试中是否遇到过这个题?
样例
样例 1:
输入: [3, 8, 4]
输出: 8
解释: 仅仅打劫第二个房子.
样例 2:
输入: [5, 2, 1, 3]
输出: 8
解释: 抢第一个和最后一个房子
动态规划求解
考虑前i项的结果dp[i]时,
dp[i]为到达第i个房间时,得到的最大收益,A为房间钱数组。
当i = 1, 返回dp[0] = A[0]
当i = 2, 返回dp[1] = max(A[0], A[1])
当i = 3, 分为偷3号房屋和不偷3号房屋,
偷的情况下, 2号房间就不能偷了,结果为A[2] + dp[0]
不偷的情况下,结果为dp[1]
所以返回dp[2] = max(dp[0] + A[2], dp[1])
以此类推,dp[i] = max(dp[i-2] + A[i], dp[i-1])
class Solution:
"""
@param A: An array of non-negative integers
@return: The maximum amount of money you can rob tonight
"""
def houseRobber(self, A):
# write your code here
if not A:
return 0
dp = [0 for _ in A]
dp[0] = A[0]
for i in range(1, len(A)):
if i == 1:
dp[i] = max(dp[0], A[i])
else:
dp[i] = max(dp[i - 2] + A[i], dp[i - 1])
return dp[-1]
c=Solution()
d=c.rob([3,8,4,13])
print(d)
输出21