[leetcode]python3 算法攻略-爬楼梯

假设你正在爬楼梯。需要 n 步你才能到达楼顶。

每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?

注意:给定 n 是一个正整数。

方案一:每一步都是前两步和前一步的和

class Solution(object):
    def climbStairs(self, n):
        """
        :type n: int
        :rtype: int
        """
        pre, cur = 1, 1
        for i in range(1,n):
            pre,cur = cur,pre+cur
        return cur

方案二:用列表记录每个n对应的值,最后的n取最后一个值即可

class Solution(object):
    def climbStairs(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n == 1:
            return 1
        if n == 2:
            return 2
        res = [1, 2]
        for i in range(2, n):
            res.append(res[i - 1] + res[i - 2])
        return res[-1]

猜你喜欢

转载自blog.csdn.net/zhenghaitian/article/details/81074773