题目
假设你正在爬楼梯,需要n个台阶才能爬到顶楼,每次你可以爬1或2个台阶,你有多少种方法可以爬到顶楼。
输入一个正整数n
输出结果正整数
思路
阶段划分
动态规划的阶段划分需要满足后无效性,当前的状态是对过去的总结。
因此将到达第i阶台阶作为第i个阶段,而阶段的状态是“到达第i阶台阶有多少种爬楼方式”,记作
d
p
[
i
]
dp[i]
dp[i]。
状态转移函数
因为有两种爬楼方式,所以达到第i阶有两种方式:① 从前一个台阶开始,爬1个台阶
② 从前两个台阶开始,爬2个台阶
由此,可以写出状态转移函数
d
p
[
i
]
dp[i]
dp[i]=
d
p
[
i
−
1
]
dp[i-1]
dp[i−1]+
d
p
[
i
−
2
]
dp[i-2]
dp[i−2]
边际条件
前两个台阶的状态是比较容易算出来的,
d
p
[
1
]
dp[1]
dp[1]=1
d
p
[
2
]
dp[2]
dp[2]=2
求解方法
顺序遍历n个台阶,
每次更新一下当前阶段的状态
代码
def solution(n):
dp = [1, 2]
for i in range(2, n):
tmp = dp[i-1] + dp[i-2]
dp.append(tmp)
return dp[n-1]