小白解读Leetcode746. 使用最小花费爬楼梯——动态规划+滚动数组
1. 核心思路
本题最核心的思路其实是要把它当做数学关系看待,其实大家都隐隐约约能感觉到跟斐波那契数列或者是爬楼梯很像。只是多了一个“花费”,所以可以肯定的是大体思路是不变的,只是具体操作需要改一点点。
2. 具体分析
① 以终为始,带着目的去想,最终是想要知道超过索引为len(cost)-1的台阶最少需要多少花费。
② 超过最后一个台阶无非也就两种方法:
第一,从len(cost) -1 再走一步(走两步也行,反正没差别,代码写的是行为结果后代表的数学运算,一步两步的运算结果都是一样的,都是加上自身台阶所需花费即可,后续也不需要再计算,所以不用纠结)。
第二,从len(cost) - 2再走两步,这里不走一步是因为cost都是正数,两步能直接到达目的地,就不要先上一阶再上一阶,这样肯定花费更多。
③ 假设走到 len(cost) - 1所需的最小花费是fmin(n - 1),走到 len(cost) - 2所需的最小花费是fmin(n - 2),那么 超过len(cost) - 1所需的最小花费 就是 :
min ( fmin (n - 1) + cost [ n - 1 ],fmin (n - 2) + cost [ n - 2 ] )
④ 根据上面的式子,可以推广到,对于 索引 n >= 2的任何一个台阶都可以用这个公式计算到达该台阶的最小花费。 所以这时候就可以想到递推了。初始值fmin(0)、fmin(1)即为到达索引为0和1台阶的最小花费,因为题目说一开始可以选择从 0 或 1 走,所以fmin(0) 、fmin(1)自然而然也应该等于0。只不过从这两个台阶离开的时候才需要加上cost而已。
⑤但是其实我们并不需要知道到达每个台阶所需要的最小花费,我们只要超过len(cost)- 1的最小花费。把这个过程想象成“摸石头过河”游戏就可以明白为什么使用滚动数组了,如下图所示(图片引自百度)。所谓滚动数组,直白点理解就是丢弃掉前面的值,只保留最新的结果。
3. 代码 (以golnag为例)
func minCostClimbingStairs(cost []int) int {
// 超过len(cost)-1就算是爬到顶部了
// 爬到 最顶层有两种方式 :len(cost) - 1 爬一步,len(cost)-2爬两步
// 设fmin(n)是爬到当前台阶所需要的最小代价,由题目可知fmin(0)、fmin(1) 都为0
// 所以,最终的代价 = min(fmin(n-2) + cost[n-2], fmin(n-1)+cost[n-1])
fmina, fminb := 0,0 // fmina,fminb是fmin(n-2)、fmin(n-1)的代表
fminc := 0 // 用于滚动数组的辅助变量
for j := 2; j <= len(cost); j++ { // 千万要注意,要等于len(cost)
fminc = min(fmina + cost[j-2], fminb + cost[j-1])
fmina = fminb
fminb = fminc
}
return fminc
}