import heapq def min_refuel_stops(n: int, stations: list[int], L: int) -> int: stops = 0 # 记录加油次数 last_pos = 0 # 上一个加油站的位置 tank = L # 当前油量 pq = [] # 定义一个小根堆，用于选择最大的油量 while True: # 当前油可以到达终点，直接返回加油次数 if tank >= n - last_pos: return stops # 遍历所有加油站，找到当前能够到达的加油站中油量最多的那一个 while stations and stations[0] <= last_pos + tank: heapq.heappush(pq, -stations.pop(0)) # 如果当前无法到达任何一个加油站，则无法到达终点，返回-1 if not pq: return -1 # 在当前能够到达的加油站中选择油量最大的那一个 tank += -heapq.heappop(pq) stops += 1 # 更新上一个加油站的位置 last_pos = stations[0] if stations else n return -1 n = 500 stations = [50, 100, 150, 200, 250, 300, 350, 400, 450, 500, 550, 600, 650, 700, 750, 800, 850, 900, 950] L = 100 print(min_refuel_stops(n, stations, L)) # 2 修改上述代码并输出

修改后的代码如下所示：

import heapq

def min_refuel_stops(n: int, stations: list[int], L: int) -> int:
    stops = 0  # 记录加油次数
    last_pos = 0  # 上一个加油站的位置
    tank = L  # 当前油量
    pq = []  # 定义一个小根堆，用于选择最大的油量
    
    while True:
        # 当前油可以到达终点，直接返回加油次数
        if tank >= n - last_pos:
            return stops
        
        # 遍历所有加油站，找到当前能够到达的加油站中油量最多的那一个
        while stations and stations[0] <= last_pos + tank:
            heapq.heappush(pq, -stations.pop(0))
        
        # 如果当前无法到达任何一个加油站，则无法到达终点，返回-1
        if not pq:
            return -1
        
        # 在当前能够到达的加油站中选择油量最大的那一个
        tank += -heapq.heappop(pq)
        stops += 1
        
        # 更新上一个加油站的位置
        last_pos = stations[0] if stations else n
    
    # 注意：此处不需要返回 -1，因为已经在循环中处理了所有情况

n = 500
stations = [50, 100, 150, 200, 250, 300, 350, 400, 450, 500, 550, 600, 650, 700, 750, 800, 850, 900, 950]
L = 100
print(min_refuel_stops(n, stations, L))  # 输出 2

修改后，当无法到达终点时，函数会直接跳出循环并返回 -1；否则会返回加油次数。同时，在最后的 return 语句中不需要返回 -1。运行代码后会输出 2，表示最少需要加两次油才能到达终点。