【LeetCode】每日一题 2024_10_7 最低加油次数（堆、贪心）

news2024/10/8 11:46:57

前言

每天和你一起刷 LeetCode 每日一题~

大家国庆节快乐呀~

LeetCode 启动！

国庆最后一天，力扣还在加油站，怕不是国庆回家路上堵车了

题目：最低加油次数

代码与解题思路

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
}