Python3 conseguir problema saque

Título original https://leetcode-cn.com/problems/house-robber/

Você é um ladrão profissional para roubar plano rua da casa. Cada quarto estão na posse de algum dinheiro, as únicas restrições que afetam roubar casa do seu vizinho é equipado com sistema anti-roubo comunicar uns com os outros, se duas casas adjacentes na mesma noite, os ladrões invadiram o sistema automaticamente alarme .
Dado um representante de cada matriz de inteiros não-negativa de quantidade de armazenamento de Habitação calculado no caso de você não toque o dispositivo de alarme, pode roubar o montante máximo a.

Exemplo 1:

输入: [1,2,3,1]
输出: 4
解释: 偷窃 1 号房屋 (金额 = 1) ，然后偷窃 3 号房屋 (金额 = 3)。
     偷窃到的最高金额 = 1 + 3 = 4 。

Exemplo 2:

输入: [2,7,9,3,1]
输出: 12
解释: 偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9)，接着偷窃 5 号房屋 (金额 = 1)。
     偷窃到的最高金额 = 2 + 9 + 1 = 12 。

A resolução de problemas:

class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)
        if n == 0:
            return 0
        if n == 1:
            return nums[0]
        if n == 2:
            return max(nums[0], nums[1])
        e0 = 0
        e1 = nums[0]
        e2 = max(nums[0], nums[1]) #动态规划思路，但是只需要3个中间值
        for i in range(3, n + 1):
            e0, e1, e2 = e1, e2, max(nums[i-1] + e1, nums[i-2] + e0)
        return e2