题目描述
在一条单车道上有 n 辆车,它们朝着同样的方向行驶。给你一个长度为 n 的数组 cars ,其中 cars[i] = [position i, speed i] ,它表示:
position i 是第 i 辆车和道路起点之间的距离(单位:米)。题目保证 position i < position i+1 。
speed i 是第 i 辆车的初始速度(单位:米/秒)。
简单起见,所有车子可以视为在数轴上移动的点。当两辆车占据同一个位置时,我们称它们相遇了。一旦两辆车相遇,它们会合并成一个车队,这个车队里的车有着同样的位置和相同的速度,速度为这个车队里 最慢 一辆车的速度。
请你返回一个数组 answer ,其中 answer[i] 是第 i 辆车与下一辆车相遇的时间(单位:秒),如果这辆车不会与下一辆车相遇,则 answer[i] 为 -1 。答案精度误差需在pow(10,-5) 以内。
思路分析
-
这里设置的三元组: {speed,index1,index2}
-
注意: 相撞后,其它车之间的相对速度是保持不变的,相撞的两个车的速度为后者的速度
-
但是,相撞车(作为一个系统)与右边的相对速度不变,但是与左边的相对速度发生改变,需要一个left数组记录左边的那个车的下标
-
关于时间的计算: 无论撞了多少次,都会根据最终添加进优先队列的那个t为标准。
Source code
- 代码在原作者基础上修改
class Solution {
public:
vector<double> getCollisionTimes(vector<vector<int>>& cars) {
int n = cars.size();
vector<double> ans(n,-1.0);
vector<int> left(n,0);
vector<bool> remove(n,false);
priority_queue<tuple<double,int,int>,vector<tuple<double,int,int>>,greater<tuple<double,int,int>>> pq;
for(int i = 0; i < n; ++i){
left[i] = i;
if(i > 0 && cars[i][1] < cars[i-1][1]){
double time=(cars[i][0] - cars[i-1][0])/(cars[i-1][1]-cars[i][1]);
pq.emplace(time,i-1,i);
}
}
while(!pq.empty()){
auto [t,x,y] = pq.top();
pq.pop();
if(remove[x] || remove[y]) continue;
left[y] = left[x];
remove[x] = true;
ans[x] = t;
if(left[x] > 0){
int prev = left[x] - 1;
if(cars[prev][1] > cars[y][1]){
double time=(cars[y][0]-cars[prev][0])/(cars[prev][1]-cars[y][1]);
pq.emplace(time,prev,y);
}
}
}
return ans;
}
};