假设你正在爬楼梯。需要 n
阶你才能到达楼顶。
每次你可以爬 1
或 2
个台阶。你有多少种不同的方法可以爬到楼顶呢?
示例 1:
输入:n = 2 输出:2 解释:有两种方法可以爬到楼顶。 1. 1 阶 + 1 阶 2. 2 阶
示例 2:
输入:n = 3 输出:3 解释:有三种方法可以爬到楼顶。 1. 1 阶 + 1 阶 + 1 阶 2. 1 阶 + 2 阶 3. 2 阶 + 1 阶
思路:另一道很经典的动态规划题,一开始直接想n可能有点绕,那就从简单的开始,若n = 1,爬到1层有几种方法呢,很明显就一种,爬1阶;若n = 2,爬到2层有几种方法呢,可以先走1阶再走1阶,或者直接走2阶,两种方法;接下去若n=3,爬到三层有几种方法呢,可以先走1阶再走1阶再走1阶,或者先走1阶再走2阶,或者先走2阶再走1阶共三种,举了三个例子我们可以先看看有没有什么规律,爬到三层你肯定会先到达一层或二层,然后再从一层或二层到达三层,因此三层的状态是可以由一层和二层的状态推出来的,因此写出递推公式dp[i] = dp[i-1] + dp[i-2],当然我们还可以再往后写四层五层去验证这个公式我们推的有没有问题。
根据这个公式和初始化,就可以得到相应的答案。
代码(Python):
class Solution(object):
def climbStairs(self,n):
dp = [0]* (n + 1) #创建dp数组
if n <= 2:
return n
dp[1] = 1 #初始化dp[1] = 1 dp[2] = 2
dp[2] = 2
for i in range(3,n+1):
dp[i] = dp[i-1] + dp[i-2] #递推公式
return dp[n]