Python上楼梯

假设一段楼梯共n(n>1)个台阶，小朋友一步最多能上3个台阶，那么小朋友上这段楼梯一共有多少种方法。

（小朋友真的累，我选择电梯）

大体思路用到了递归，假如说楼梯有12阶，那么11阶时有只有一种上方 12阶的时候有两种上方 13阶的时候有4中上法，递归到第一阶，第二阶 第三阶

n = int(input())
a = 1#第一层台阶只有一种上法
b = 2#第二层台阶有两种上法
c = 4#第三层台阶有4中上法
for i in range(n-3):
    m=a+b+c
    a=b
    b=c
    c=m
print(c)