题干
给你一个整数数组 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 。
解题思路
本题依旧是动态规划,用五部曲来解答。
1.确定dp数组以及下标的含义
dp[i]记录到达第i阶阶梯的最低花费
2.确定递推公式
可以有两个途径得到dp[i],一个是dp[i-1] 一个是dp[i-2]。
dp[i - 1] 跳到 dp[i] 需要花费 dp[i - 1] + cost[i - 1]。
dp[i - 2] 跳到 dp[i] 需要花费 dp[i - 2] + cost[i - 2]。
那么究竟是选从dp[i - 1]跳还是从dp[i - 2]跳呢?
一定是选最小的,所以dp[i] = min(dp[i - 1] + cost[i - 1], dp[i - 2] + cost[i - 2]);
3.确定初始值
由题意,我们可以选择从下标为 0
或下标为 1
的台阶开始爬楼梯。
因此,初始值为0。
dp[0] = 0;
dp[1] = 0;
4.确定遍历顺序
从前向后
5.检查dp数组
以cost = [1,100,1,1,1,100,1,1,100,1]为例,
dp数组的值应该为dp[2] = 1 dp[3] =2, dp[4] = 2……
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
//获得数组大小
int N = cost.size();
//建立dp数组,dp[i]记录到达第i阶阶梯的最低花费
int dp[N + 1];
//设立初始值
dp[0] = 0;
dp[1] = 0;
for(int i = 2; i <= N; i++){
//递推公式
dp[i] = min(dp[i - 1] + cost[i - 1] , dp[i - 2] + cost[i - 2]);
}
return dp[N];
}
};