题目描述
数组的每个索引作为一个阶梯,第 i个阶梯对应着一个非负数的体力花费值 cost[i](索引从0开始)。
每当你爬上一个阶梯你都要花费对应的体力花费值,然后你可以选择继续爬一个阶梯或者爬两个阶梯。
您需要找到达到楼层顶部的最低花费。在开始时,你可以选择从索引为 0 或 1 的元素作为初始阶梯。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-cost-climbing-stairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例
示例 1:
输入: cost = [10, 15, 20]
输出: 15
解释: 最低花费是从cost[1]开始,然后走两步即可到阶梯顶,一共花费15。
示例 2:输入: cost = [1, 100, 1, 1, 1, 100, 1, 1, 100, 1]
输出: 6
解释: 最低花费方式是从cost[0]开始,逐个经过那些1,跳过cost[3],一共花费6。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-cost-climbing-stairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
典型的dp问题,与爬楼梯、青蛙跳等不同的是,可以从第一步开始,也可以从第二步开始
代码
前面加上两个 0,就不用理会到底是从 位置0 开始,还是从 位置1 开始, 结尾加上一个 0,就不用理会从到 -1 结束,还是到 -2 结束。
class Solution:
def minCostClimbingStairs(self, cost: List[int]) -> int:
cost = [0, 0] + cost + [0]
dp = [0] * (len(cost))
for i in range(2, len(cost)):
dp[i] = min(dp[i-1], dp[i-2]) + cost[i]
return dp[-1]
复杂度分析
- 时间复杂度:O(n)
- 空间复杂度:O(n)