Python3 escaleras de tres pasos preguntas

Dirección del título original https://leetcode-cn.com/problems/three-steps-problem-lcci/

tema:

problemas de tres pasos. Hay un niño está en la escalera, un nivel de orden n, un niño puede estar en el orden de 1, 2 y 3 de la orden. Implementar un método para calcular cuántas maneras hay niños en las escaleras. Los resultados pueden ser muy grandes, es necesario morir 1,000,000,007 resultados.

Ejemplo 1:

 输入：n = 3 
 输出：4
 说明: 有四种走法

Ejemplo 2:

 输入：n = 5
 输出：13

La resolución de problemas:

class Solution:
    def waysToStep(self, n: int) -> int:
        if n==0:
            return 0
        if n==1:
            return 1
        if n==2:
            return 2
        if n==3:
            return 4
        # 本题用动态规划的思路解题，但是不需要n个数组空间，只需3个记录
        e1 = 1
        e2 = 2
        e3 = 4
        for i in range(4, n+1):
            e4 = (e1 + e2 + e3)%1000000007 # 注意！取模一定要在这里就算好，而不是放到最后一步
            e1 = e2
            e2 = e3
            e3 = e4
        return e3