开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第30天,点击查看活动详情
一、题目描述:
给你一个整数数组 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 。
提示:
- 2 <= cost.length <= 1000
- 0 <= cost[i] <= 999
二、思路分析:
这题是第70题爬楼梯的进阶版。也是采用动态规划的方法,将上一步的结果保存在数组中,进行判断。不同的是,这题新增了一个cost变量。
爬楼梯那题的动态规划方程为:f(n) = f(n-1)+f(n-2)
这一题的我们设定f(n)为第一步需要从索引为0的楼梯开始爬,也就是第一步一定是一步。最后返回结果的时候,返回min(f(n), f(n-1))即可
按照上述描述,该题的动态规划方程为:
f(n) = min(cost(n) + f(n-1), cost(n) + f(n-2))
也就是第二步是爬一步还是两步的问题
三、AC 代码:
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
int len = cost.size();
if(len == 0 || len == 1) return 0;
int res;
int dp[len];
dp[0] = cost[len-1];
dp[1] = cost[len-2];
for(int i = 2; i < len; i++)
{
dp[i] = min(cost[len-i-1] + dp[i-1], cost[len-i-1] + dp[i-2]);
}
return min(dp[len-1], dp[len-2]);
}
};