给你一个整数数组 cost
,其中 cost[i]
是从楼梯第 i
个台阶向上爬需要支付的费用。一旦你支付此费用,即可选择向上爬一个或者两个台阶。
你可以选择从下标为 0
或下标为 1
的台阶开始爬楼梯。
请你计算并返回达到楼梯顶部的最低花费。
示例 1:
输入:cost = [10,15,20] 输出:15 解释:你将从下标为 1 的台阶开始。 - 支付 15 ,向上爬两个台阶,到达楼梯顶部。 总花费为 15 。
示例 2:
输入: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 。
思路:这道题在爬楼梯的基础上增加了爬楼梯的代价,让我们给出最小的爬楼代价。
首先大体方向跟爬楼梯不会变,爬到n层是由爬到n-1层和n-2层推导决定,也就是dp[i] = dp[i-1] + dp[i-2],但这题引入了cost花费概念,上面的推导公式是求出爬到n层有几种爬的方式,但我们在这里只需要选择一种方式,且花费最小,因此改进递推公式为 dp[i] = min(dp[i-1]+cost[i-1],dp[i-2]+cost[i-2])。然后确定初始值,因为一开始是从0或1开始跳,到0或1是不需要花费代价的,因此初始值dp[0]和dp[1]都是0.
代码(Python):
class Solution(object):
def minCostClimbingStairs(self, cost):
dp = [0] * (len(cost) + 1)
dp[0] = 0
dp[1] = 0
for i in range(2,len(cost)+1):
dp[i] = min(dp[i-1]+cost[i-1],dp[i-2]+cost[i-2]) #递推公式
return dp[len(cost)]