LeetCode 392打劫房屋 python

描述
假设你是一个专业的窃贼，准备沿着一条街打劫房屋。每个房子都存放着特定金额的钱。你面临的唯一约束条件是：相邻的房子装着相互联系的防盗系统，且 当相邻的两个房子同一天被打劫时，该系统会自动报警。

给定一个非负整数列表，表示每个房子中存放的钱， 算一算，如果今晚去打劫，在不触动报警装置的情况下, 你最多可以得到多少钱 。

您在真实的面试中是否遇到过这个题？
样例
样例 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