前言
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/climbing-stairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
- 1 阶 + 1 阶 + 1 阶
- 1 阶 + 2 阶
- 2 阶 + 1 阶
思路
这道题可以使用递归的思想来解答。
首先,确定递归结束条件,就是当只有1个台阶或2个台阶时,输出就是1或者2。所以递归结束条件就是n<=2。
其次,找出函数的等价关系,每次跳的时候,可以跳一个台阶,也可以跳两个台阶。第一种跳法:第一次跳了一个台阶,还剩下n-1个台阶还没跳,剩下的n-1个台阶的跳法有f(n-1)种。第二种跳法:第一次跳了两个台阶,还剩下n-2个台阶还没,剩下的n-2个台阶的跳法有f(n-2)种。所以,全部跳法就是这两种跳法之和了,即 f(n) = f(n-1) + f(n-2)。
解答
虽然做出来了,但是运行超出时间限制了,还需要优化一下。
class Solution:
def climbStairs(self, n):
if n <= 2:
return n
return Solution.climbStairs(self, n - 1) + Solution.climbStairs(self, n - 2)
if __name__ == '__main__':
res = Solution().climbStairs(3)
print(res)