其他分享
首页 > 其他分享> > 车队 II

车队 II

作者:互联网

题目描述

在一条单车道上有 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) 以内

思路分析 

  1. 这里设置的三元组: {speed,index1,index2}

  2. 注意: 相撞后,其它车之间的相对速度是保持不变的,相撞的两个车的速度为后者的速度

  3. 但是,相撞车(作为一个系统)与右边的相对速度不变,但是与左边的相对速度发生改变,需要一个left数组记录左边的那个车的下标

  4. 关于时间的计算: 无论撞了多少次,都会根据最终添加进优先队列的那个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;
    }
};

标签:pq,cars,remove,车队,II,vector,prev,left
来源: https://blog.csdn.net/qq_36822894/article/details/121165768