Leetcode 853. 车队 C++

Leetcode 853. 车队

题目

N 辆车沿着一条车道驶向位于 target 英里之外的共同目的地。

每辆车 i 以恒定的速度 speed[i] （英里/小时），从初始位置 position[i] （英里） 沿车道驶向目的地。

一辆车永远不会超过前面的另一辆车，但它可以追上去，并与前车以相同的速度紧接着行驶。

此时，我们会忽略这两辆车之间的距离，也就是说，它们被假定处于相同的位置。

车队 是一些由行驶在相同位置、具有相同速度的车组成的非空集合。注意，一辆车也可以是一个车队。

即便一辆车在目的地才赶上了一个车队，它们仍然会被视作是同一个车队。

会有多少车队到达目的地?

示例：

输入：target = 12, position = [10,8,0,5,3], speed = [2,4,1,1,3]
输出：3
解释：
从 10 和 8 开始的车会组成一个车队，它们在 12 处相遇。
从 0 处开始的车无法追上其它车，所以它自己就是一个车队。
从 5 和 3 开始的车会组成一个车队，它们在 6 处相遇。
请注意，在到达目的地之前没有其它车会遇到这些车队，所以答案是 3。

提示：

1. 0 <= N <= 10 ^ 4
2. 0 < target <= 10 ^ 6
3. 0 < speed[i] <= 10 ^ 6
4. 0 <=position[i] < target
5. 所有车的初始位置各不相同。

题解

先假设后车追上前车时不会改变速度，这样我们能得到每辆车到达终点的时间。
我们思考以下，车队是怎么汇集的，显然是因为当前车的前面有比它晚到终点的。利用这个关系，我们可以维持一个单调递减栈，其中栈元素为车到达终点的时间。最终的答案就是栈中元素个数。
注意一点是，我们首先要将车根据起点的位置进行排序，保证前后车的关系；也就是可以根据距终点的距离进行逆序排序。详细过程见代码

代码

bool sortP(vector<int>& p1,vector<int>& p2){
    return p1[0] > p2[0];
}
class Solution {
public:
    int carFleet(int target, vector<int>& position, vector<int>& speed) {
        int n = position.size();
        if(n == 1)  return 1;
        vector<vector<int>> cars;
        for(int i=0; i<n; i++)		//获取终点的距离
            cars.push_back({target-position[i],speed[i]});
        sort(cars.begin(),cars.end(),sortP);		//根据距终点的距离进行逆序排序，保证前后车关系
        stack<double> sp;
        for(int i=0; i<n; i++){
            double time = (double)cars[i][0]/cars[i][1];		//其到达终点的时间
            while(!sp.empty() && time>=sp.top())		//维持单调递减栈
                sp.pop();
            sp.push(time);
        }
        return sp.size();
    }
};

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/car-fleet
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。