LeetCode刷题3--爬楼梯

爬楼梯

链接

题目描述

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

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

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

示例 1：

输入： 2
输出： 2
解释： 有两种方法可以爬到楼顶。

1. 1 阶 + 1 阶
2. 2 阶

示例 2：

输入： 3
输出： 3
解释： 有三种方法可以爬到楼顶。

1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶

思路

思路：
f[0]=0
f[1]=1
n=2时：可以1+1或者 2，因此有两种
n=3时：横着放还剩 f[2],竖着放还剩f[1]
因此 f[n]=f[n-1]+f[n-2]

代码

非递归方法

##### 非递归方法
class Solution(object):
    def climbStairs(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n == 0:
            return 0
        f_1 = 0
        f_2 = 1
        for i in range(1, n + 1):
            f_n = f_1 + f_2
            f_1 = f_2
            f_2 = f_n
        return f_n

递归方式，会超时

class Solution(object):
    def climbStairs(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n == 0:
            return 0
        if n == 1:
            return 1
        if n == 2:
            return 2
        f_n = self.climbStairs(n - 1) + self.climbStairs(n - 2)
        return f_n