欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

lintcode:爬楼梯(递归,迭代,动态规划)

程序员文章站 2022-03-24 17:37:38
...

lintcode:爬楼梯(递归,迭代,动态规划)

class Solution:
    """
    @param n: An integer
    @return: An integer
    """
    def climbStairs(self, n):
        # write your code here
        # n 1 2 3 4 5
        # r 1 2 3 5 8

        # 递归 超时 70% 39
        # if n == 1:
        #     return 1
        # if n == 2:
        #     return 2
        # return self.climbStairs(n - 1) + self.climbStairs(n - 2)

        # 迭代
        # a, b = 1, 2
        # for i in range(1, n):
        #     a, b = b, a + b
        # return a

        # 动态规划
        dp = [0]*(n + 1)
        if n == 0 or n == 1:
            return 1
        dp[1], dp[2] = 1, 2
        for i in range(3, n + 1):
            dp[i] = dp[i - 1] + dp[i - 2]
            # print(i)
            # print(dp[i - 1])
        return dp[n]

刚开始用动态规划时开的数组是n 导致循环时一直越界,一直在改range和return中的n, 结果就是一直错,后来才改的数组大小。