「动态规划」LeetCode 70(爬楼梯)
Leetcode 70 题
有人问我:烤冷面你这两周怎么总搞简单题?我想说:一步一步来~
题干简述
给定:
- 假设你正在爬楼梯,需要爬 n 阶你才能到达楼顶。
- 每次你可以爬 1 或 2 个台阶。
要求:计算出有多少种爬楼梯的方式。
解题思路
如果我们缩小视野(把大问题化为小问题),爬到第 n 阶台阶有两种方式:
- 从 n-1 阶爬一级台阶
- 从 n-2 阶爬两级台阶
用公式表达:dp[n] = dp[n−1] + dp[n−2]
,其中的特例是:dp[0]=1 和 dp[1]=1
。
嚯!这不就是LeetCode 509(斐波那契数列)么。
代码实现
class Solution:
def climbStairs(self, n: int) -> int:
if n <= 2:
return n
dp = [0]*(n+1)
dp[0] = 1
dp[1] = 1
for i in range(2, n+1):
dp[i] = dp[i-1] + dp[i-2]
return dp[n]
复杂度
时间复杂度O(n),空间复杂度O(n)。
关注公众号:「腐蚀脚本」,加入交流群。
文章归档:烤冷面讲算法系列
转载声明:本文章允许转载,原文地址:「动态规划」LeetCode 70(爬楼梯)