题目
给你一个整数数组 cost ,其中 cost[i] 是从楼梯第 i 个台阶向上爬需要支付的费用。一旦你支付此费用,即可选择向上爬一个或者两个台阶。
你可以选择从下标为 0 或下标为 1 的台阶开始爬楼梯。
请你计算并返回达到楼梯顶部的最低花费。
示例 1:
输入:cost = [10,15,20]
输出:15
解释:你将从下标为 1 的台阶开始。
- 支付 15 ,向上爬两个台阶,到达楼梯顶部。
总花费为 15 。
思路
到达楼顶有两种方式:
- 在最后一级台阶爬一步;
- 在倒数第二级台阶爬两步。
所以,到达楼顶的最小成本应当是两者中的较小值,属于动态规划问题。
我们假设第i级台阶的最小成本为dp[i],那么:
dp[i] = min(dp[i - 1] + cost[i - 1], dp[i - 2] + cost[i - 2]);
题目中明确描述:
你可以选择从下标为 0 或下标为 1 的台阶开始爬楼梯
因此,在第0级和第1级台阶上不需要成本。即dp[0]=dp[1]=0.
细节
dp的长度是需要着重考虑的。
请你计算并返回达到楼梯顶部的最低花费。
这里强调达到楼梯顶部,如果仅仅是达到最后一级台阶,是不符合题意的。
因此dp的长度应该在楼梯级数的基础上再加1,dp数组的最后一个值才对应“楼顶”,而倒数第一个值对应“最后一级阶梯”
代码
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
int len = cost.size();
vector<int> dp(len+1, 0);//dp的长度应该在楼梯级数的基础上再加1
dp[0] = 0;
dp[1] = 0;
for (int i = 2; i < len+1; i++){
dp[i] = min(dp[i - 1] + cost[i - 1], dp[i - 2] + cost[i - 2]);
}
return dp[len];//dp数组的最后一个值才对应“楼顶”
}
};