1.题目描述
给你一个整数数组
cost
,其中cost[i]
是从楼梯第i
个台阶向上爬需要支付的费用。一旦你支付此费用,即可选择向上爬一个或者两个台阶。你可以选择从下标为
0
或下标为1
的台阶开始爬楼梯。请你计算并返回达到楼梯顶部的最低花费。
输入:cost = [10,15,20]
输出:15
解释:你将从下标为 1 的台阶开始。
- 支付 15 ,向上爬两个台阶,到达楼梯顶部。
总花费为 15 。输入:cost = [1,100,1,1,1,100,1,1,100,1]
输出:6
解释:你将从下标为 0 的台阶开始。
- 支付 1 ,向上爬两个台阶,到达下标为 2 的台阶。
- 支付 1 ,向上爬两个台阶,到达下标为 4 的台阶。
- 支付 1 ,向上爬两个台阶,到达下标为 6 的台阶。
- 支付 1 ,向上爬一个台阶,到达下标为 7 的台阶。
- 支付 1 ,向上爬两个台阶,到达下标为 9 的台阶。
- 支付 1 ,向上爬一个台阶,到达楼梯顶部。
总花费为 6 。提示:
2 <= cost.length <= 1000
0 <= cost[i] <= 999
2.思路分析
假设数组 cost 的长度为 n,则 n 个阶梯分别对应下标 0 到 n-1,楼层顶部对应下标 n,问题等价于计算达到下标 n 的最小花费。
-
确定dp数组以及下标含义
- dp[i]的定义:到达第i个台阶所花费的最少花费。
-
确定递推公式
- **状态转移方程 dp[i]=min(dp[i−1]+cost[i−1],dp[i−2]+cost[i−2]) **
-
dp数组的初始化
- 由于可以选择下标 0 或 1 作为初始阶梯,因此有 dp[0]=dp[1]=0。
-
确定遍历顺序
- 前->后(根据递推公式可以看出)
-
举例推导dp数组(n=5)
cost = [1, 100, 1, 1, 1, 100, 1, 1, 100, 1]
3.代码实现
class Solution:
def minCostClimbingStairs(self, cost: List[int]) -> int:
n = len(cost)
dp = [0] * (n + 1)
for i in range(2, n + 1):
dp[i] = min(dp[i - 1] + cost[i - 1], dp[i - 2] + cost[i - 2])
return dp[n]
复杂度分析
- 时间复杂度:O(n) 。
- 空间复杂度:O(n)。
class Solution:
def minCostClimbingStairs(self, cost: List[int]) -> int:
n = len(cost)
prev = curr = 0
for i in range(2, n + 1):
nxt = min(curr + cost[i - 1], prev + cost[i - 2])
prev, curr = curr, nxt
return curr
复杂度分析
- 时间复杂度:O(n) 。
- 空间复杂度:O(1)。