【Python】【难度：简单】Leetcode 面试题 08.01. 三步问题

三步问题。有个小孩正在上楼梯，楼梯有n阶台阶，小孩一次可以上1阶、2阶或3阶。实现一种方法，计算小孩有多少种上楼梯的方式。结果可能很大，你需要对结果模1000000007。

示例1:

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

 输入：n = 5
 输出：13
提示:

n范围在[1, 1000000]之间

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/three-steps-problem-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

class Solution(object):
    def waysToStep(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n==1:
            return 1
        if n==2:
            return 2
        if n==3:
            return 4

        var=[0]*(n+1)

        var[0]=0
        var[1]=1
        var[2]=2
        var[3]=4

        for i in range(4,n+1):
            var[i]=(var[i-1]+var[i-2]+var[i-3])%1000000007

        return var[n]

 

执行结果：

通过

显示详情

执行用时 :380 ms, 在所有 Python 提交中击败了29.21%的用户

内存消耗 :71.6 MB, 在所有 Python 提交中击败了100.00%的用户