【Leetcode】1776. Car Fleet II

题目地址：

https://leetcode.com/problems/car-fleet-ii/description/

有$n$辆车都沿着数轴向右做同向运动，给出每个车的位置和速度，位置从小到大给出。如果后车追尾了前车，其会合并进前车中并随着前车一起继续走。初始时间为$0$，问每个车的追尾时间。如果某个车没追尾则规定答案为$-1$。题目保证每个车的初始位置不重合。

设每个车的位置和速度分别是$x_i$和$v_i$，我们将$(x_i,v_i)$看成是平面直角坐标系中的若干个点，则这些点已经按横坐标排好序。例如我们考虑第$i$个车，那么这个车与之后的车追尾的时间应该是 min ⁡ { x j − x i v i − v j : j > i ∧ v i > v j } \min\{\frac{x_j-x_i}{v_i-v_j}:j>i\land v_i>v_j\} min{vi​−vj​xj​−xi​​:j>i∧vi​>vj​}，那我们也就是要求 min ⁡ { v j − v i x j − x j : j > i ∧ v i > v j } \min\{\frac{v_j-v_i}{x_j-x_j}:j>i\land v_i>v_j\} min{xj​−xj​vj​−vi​​:j>i∧vi​>vj​}，在图上看，其实就是要求所有$(x_i,v_i)$右侧的并且在其水平线下方的点中，哪个点与这个点连起来的直线斜率最小（斜率一定是负的，直观上看就是要这个直线尽量“竖直”）。很显然，要求的点一定在候选点的凸包上，凸包求法参考https://blog.csdn.net/qq_46105170/article/details/127714131。由于我们只需要看凸包下半部分的点，所以我们是倒着遍历。考虑新点$p$的时候，我们看一下$p$和栈顶两个元素$p_1,p_2$（$p_2$是栈顶，$p_1$是栈顶下面那个元素），如果$\overrightarrow{p{p}_1}$和$\overrightarrow{p{p}_2}$成右手系，则$p_2$既不是要找的点，也不在凸包上，从而可以出栈。如此这般继续操作，如果栈空了，说明要么后面没有点了，要么速度都更快（或者等速），那么其不会追尾，答案为$-1$；如果栈不空，看一下得到的点的速度，如果其速度大于等于当前点，则也不会追尾；否则会追尾，求一下追尾时间即可。代码如下：

class Solution {
 public:
  using PDD = pair<double, double>;
#define x first
#define y second

#define cross(x1, y1, x2, y2) ((x1) * (y2) - (x2) * (y1))

  double area(PDD &a, PDD &b, PDD &c) {
    return cross(b.x - a.x, b.y - a.y, c.x - a.x, c.y - a.y);
  }

  vector<double> getCollisionTimes(vector<vector<int>> &v) {
    int n = v.size();
    vector<PDD> stk;
    vector<double> res(n);
    for (int i = n - 1; i >= 0; i--) {
      auto &c = v[i];
      PDD p(c[0], c[1]);
      while (stk.size() >= 2 && area(p, stk[stk.size() - 2], stk.back()) >= 0)
        stk.pop_back();
      if (stk.empty()) res[i] = -1;
      else {
        auto &q = stk.back();
        if (p.y <= q.y) res[i] = -1;
        else res[i] = (q.x - p.x) / (p.y - q.y);
      }

      stk.push_back(p);
    }

    return res;
  }
};

时空复杂度$O(n)$。