392. 打劫房屋

392. 打劫房屋

中文English

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

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

样例

样例 1:

输入: [3, 8, 4]
输出: 8
解释: 仅仅打劫第二个房子.

样例 2:

输入: [5, 2, 1, 3] 
输出: 8
解释: 抢第一个和最后一个房子

挑战

O(n) 时间复杂度 且 O(1) 存储。

输入测试数据 (每行一个参数)如何理解测试数据？

第一种写法：时间复杂度O(n)，空间复杂度O(n)

class Solution:
    """
    @param A: An array of non-negative integers
    @return: The maximum amount of money you can rob tonight
    """
    def houseRobber(self, A):
        if not A:
            return 0
        
        if len(A) <= 2:
            return max(A)
        
        #如果是长度大于3的话
        f = [A[0],max(A[:2])]
        for i in range(2,len(A)):
            #根据上一个 和 前两个 + 当前值 进行比较
            f.append(max(f[i-1],f[i-2] + A[i]))
        return f[-1]
#result = Solution().houseRobber([5,2,1,3])
#print(result)

滚动数组方式：时间复杂度O(n),空间复杂度O(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):
        if not A:
            return 0
        
        if len(A) <=2 :
            return max(A)
        
        f = [0]*3
        f[0],f[1] = A[0],max(A[:2])
        for i in range(2,len(A)):
            #滚动的方式进行赋值，每次赋值最新的
            f[i%3] = max(f[(i-1)%3],f[(i-2)%3] + A[i])
        return f[(len(A)-1)%3]
result = Solution().houseRobber([5,2,1,3])
print(result)