题目描述:
给你一个整数数组 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 。
解题思路:
这是一个动态规划问题。设 dp[i] 表示爬到第 i 个台阶所需的最小费用,那么有以下转移方程:dp[i] = min(dp[i-1], dp[i-2]) + cost[i]
其中,dp[0]=cost[0],dp[1]=cost[1]。最终的答案即为 dp[n-1] 和 dp[n-2] 中较小的值,其中 n 是台阶数。
解释:考虑到每次可以选择向上爬一个或者两个台阶,因此到达第 i 个台阶时,前一步只能是从第 i-1 或者 i-2 个台阶来的。而在这两种情况中,选取最小花费的路径即可,因为要求的是最低花费。最后比较 dp[n-1] 和 dp[n-2] 的值,选择较小的一个作为最终答案。
代码如下:
const getMinSum = function (nums) {
// 思考:到达最后一个楼梯只有两种方法,一个上一层楼梯,另外一个是上两层楼梯,找到其中花费最少的方法
// 是一个动态规划的问题
let len = nums.length;
// 如果楼梯只有一层的情况
if (len === 1) return nums[0];
// 如果楼梯只有二层的情况
if (len === 2) return Math.min(nums[0], nums[1]);
// 定义记忆数组来保存到达每层的最少花费
let dep0 = nums[0],
dep1 = nums[1];
let dep = [dep0, dep1];
// 循环求解
for (let i = 2; i < len; i++) {
// 这个地方与爬楼梯区别:就是不要忘记后续加上当时楼梯所花费的费用
dep[i] = Math.min(dep[i - 1], dep[i - 2]) + nums[i];
}
return Math.min(dep(len - 1), dep(len - 2));
};