题目地址:
https://leetcode.com/problems/minimum-number-of-refueling-stops/description/
有一个小车,初始位置在 0 0 0,它一路向右走,给定若干位置,每个位置有加油站可以加一定数量的油,耗油和距离是 1 : 1 1:1 1:1的关系。再给定一个目标位置 t t t,问小车能否到达 t t t。如果到达 t t t的时候油恰好耗光也是可以的。
我们可以先忽略所有的加油站一直向右走,如果发现走不到 t t t,则挑选经过过的加油站里油最多的那个加油继续走,如此循环。如果发现没走到 t t t但是用光了备选的加油站,则说明到不了。代码如下:
class Solution {
public:
int minRefuelStops(int t, int fuel, vector<vector<int>>& ss) {
int res = 0, x = fuel;
priority_queue<int> heap;
auto it = ss.begin();
while (x < t) {
// 把已经经过的加油站的油都入堆
while (it != ss.end() && it->at(0) <= x) {
heap.push(it->at(1));
it++;
}
// 如果没有备选加油站了,则说明到不了
if (heap.empty()) return -1;
// 选那个油最多的加油站加油
x += heap.top(); heap.pop();
res++;
}
return res;
}
};
时间复杂度 O ( n log n ) O(n\log n) O(nlogn), n n n为加油站数量,空间 O ( n ) O(n) O(n)。