题目描述(简单)
假设你正在爬楼梯。需要 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 阶
思路分析
此题可以使用递归和动态规划两种方法,但是递归方法时间复杂度过高,会运行超时。
两种方法的思想都是要考虑前面楼梯的方法数。比如想到到达第 n 个楼梯,则前面一步可能是从 n-1, 或者 n-2 的楼梯处爬来的(因为每次只能爬1或者2阶楼梯),用dp[n]表示爬到 n 个楼梯的方法数,这一思路即: dp[n]=dp[n-1]+dp[n-2]。一直到边界条件dp[0]=1,dp[1]=2。
代码
class Solution:
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
##以下是递归方法
# if n==1:
# return 1
# elif n==2:
# return 2
# return self.climbStairs(n-1)+self.climbStairs(n-2)
##以下是动态规划
dp=[1]*n
dp[1]=2
for i in range(2,n):
dp[i]=dp[i-1]+dp[i-2]
return dp[n-1]