前言
每天和你一起刷 LeetCode 每日一题~
大家国庆节快乐呀~
LeetCode 启动!
国庆最后一天,力扣还在加油站,怕不是国庆回家路上堵车了
题目:最低加油次数
代码与解题思路
go
func minRefuelStops(target int, startFuel int, stations [][]int) (ans int) {
// 题目分析:
// 1、target 是目的地的距离,startFuel 是初始油量,stations 是加油站
// 2、stations 加油站的距离列表是升序列表,不需要我们排序
// 也就意味着,我们只需要直接遍历加油站列表,就能走完所有的加油站,直到目的地 target
// 3、题目要求我们记录要加多少次的油,贪心的想,每次都加油最多的那个加油站
// 我们将当前的油量能抵达的加油站都入最大堆,维护加油量最大的加油站,油不够了就加最大的那个
stations = append(stations, []int{target, 0}) // 最后一站 target
curFuel := startFuel // 当前的油量
fuelHeap := &hp{}
// 遍历 stations,如果能走到最后一站油箱里还有油,就证明可以抵达
for _, st := range stations {
pos, fuel := st[0], st[1] // 每一站的距离和能加的油
fmt.Println(curFuel, " ", pos, " ", fuelHeap.Len()) // 发现 fuelHeap 没 push 进去
for fuelHeap.Len() > 0 && curFuel < pos { // 有经过的加油站,需要加油且有油能加
curFuel += heap.Pop(fuelHeap).(int)
ans++ // 加油次数++
}
if curFuel < pos { // 走不到下一个加油站,返回 -1
return -1
}
heap.Push(fuelHeap, fuel) // 这一站能走这么远,把这个加油站收入堆中
}
return ans
}
type hp struct{ sort.IntSlice }
func (h hp) Less(i, j int) bool { // 默认是最小堆,我们设置成最大堆
return h.IntSlice[i] > h.IntSlice[j]
}
func (h *hp) Push(v any) {
h.IntSlice = append(h.IntSlice, v.(int))
}
func (h *hp) Pop() any {
a := h.IntSlice
v := a[len(a)-1]
h.IntSlice = a[:len(a)-1]
return v
}
详细思路如注释
这道题目的思路的核心就在于:
在当前油量能抵达的所有油站中,选择油量最大的油站加油,直到走到终点。怎么样才能维护油量最大的油站?那就需要用到最大堆。
视频实况
【【LeetCode】每日一题 2024_10_7 最低加油次数(堆、贪心)】
每天进步一点点
可以和我刷一辈子的每日一题吗?
一题一题,积累起来就是一辈子。