数组题目：所有蚂蚁掉下来前的最后一刻

题目

标题和出处

标题：所有蚂蚁掉下来前的最后一刻

出处：1503. 所有蚂蚁掉下来前的最后一刻

难度

5 级

题目描述

要求

有一块木板，长度为 $\text{n}$ 个单位。一些蚂蚁在木板上移动，每只蚂蚁都以每秒一个单位的速度移动。其中，一部分蚂蚁向左移动，其他蚂蚁向右移动。

当两只向不同方向移动的蚂蚁在某个点相遇时，它们会同时改变移动方向并继续移动。假设更改方向不会花费任何额外时间。

而当蚂蚁在某一时刻 $\text{t}$ 到达木板的一端时，它立即从木板上掉下来。

给你一个整数 $\text{n}$ 和两个整数数组 $\text{left}$ 以及 $\text{right}$。两个数组分别标识向左或者向右移动的蚂蚁在 $\text{t = 0}$ 时的位置。请你返回最后一只蚂蚁从木板上掉下来的时刻。

示例

示例 1：

输入：$\text{n = 4, left = [4,3], right = [0,1]}$
输出：$\text{4}$
解释：如上图所示：

• 下标 $\text{0}$ 处的蚂蚁命名为 $\text{A}$ 并向右移动。
• 下标 $\text{1}$ 处的蚂蚁命名为 $\text{B}$ 并向右移动。
• 下标 $\text{3}$ 处的蚂蚁命名为 $\text{C}$ 并向左移动。
• 下标 $\text{4}$ 处的蚂蚁命名为 $\text{D}$ 并向左移动。

请注意，蚂蚁在木板上的最后时刻是 $\text{t = 4}$ 秒，之后蚂蚁立即从木板上掉下来。（也就是说在 $\text{t = 4.0000000001}$ 时，木板上没有蚂蚁）。

示例 2：

输入：$\text{n = 7, left = [], right = [0,1,2,3,4,5,6,7]}$
输出：$\text{7}$
解释：所有蚂蚁都向右移动，下标为 $\text{0}$ 的蚂蚁需要 $\text{7}$ 秒才能从木板上掉落。

示例 3：

输入：$\text{n = 7, left = [0,1,2,3,4,5,6,7], right = []}$
输出：$\text{7}$
解释：所有蚂蚁都向左移动，下标为 $\text{7}$ 的蚂蚁需要 $\text{7}$ 秒才能从木板上掉落。

示例 4：

输入：$\text{n = 9, left = [5], right = [4]}$
输出：$\text{5}$
解释：$\text{t = 1}$ 秒时，两只蚂蚁将回到初始位置，但移动方向与之前相反。

示例 5：

输入：$\text{n = 6, left = [6], right = [0]}$
输出：$\text{6}$

数据范围

• $\text{1}\le \text{n}\le {\text{10}}^{\text{4}}$
• $\text{0}\le \text{left.length}\le \text{n + 1}$
• $\text{0}\le \text{left[i]}\le \text{n}$
• $\text{0}\le \text{right.length}\le \text{n + 1}$
• $\text{0}\le \text{right[i]}\le \text{n}$
• $\text{1}\le \text{left.length + right.length}\le \text{n + 1}$
• $\text{left}$ 和 $\text{right}$ 中的所有值都是唯一的，并且每个值只能出现在二者之一中

解法

思路和算法

题目描述中的「当两只向不同方向移动的蚂蚁在某个点相遇时，它们会同时改变移动方向并继续移动」非常具有迷惑性。根据常规思路，需要维护每只蚂蚁的位置和移动方向，当两只移动方向不同的蚂蚁相遇时，更新两只蚂蚁的移动方向。这种做法需要计算每次出现蚂蚁相遇的时间，因此实现是非常麻烦的。

有没有其他实现方法？考虑以下情形：假设有两只蚂蚁，$A$ 的位置是 $a$，$B$ 的位置是 $b$，其中 $a<b$，$A$ 向右移动，$B$ 向左移动，它们会在位置 $c=\frac{a+b}{2}$ 相遇，相遇之后，两只蚂蚁同时改变移动方向，即 $A$ 向左移动，$B$ 向右移动。

上述情形中，从相遇的时刻开始过了时间 $\Delta t$ 之后，$A$ 的位置是 $c-\Delta t$，$B$ 的位置是 $c+\Delta t$。由于每只蚂蚁的移动速度都是相同的，因此如果 $A$ 和 $B$ 相遇之后按照原来的移动方向继续移动，则从相遇的时刻开始过了时间 $\Delta t$ 之后，$A$ 的位置是 $c+\Delta t$，$B$ 的位置是 $c-\Delta t$。注意到两种情况的区别在于将 $A$ 和 $B$ 进行了交换。由于这道题只需要知道最后一只蚂蚁从木板上掉下来的时刻，不需要知道每只蚂蚁从木板上掉下来的时刻，因此可以将蚂蚁移动的过程等价地转化成两只蚂蚁相遇后按照原来的移动方向继续移动。

如果一只蚂蚁的初始位置是 $x$，则当这只蚂蚁向左移动时，从木板上掉下来的时刻是 $x$，当这只蚂蚁向右移动时，从木板上掉下来的时刻是 $n-x$。

遍历数组 $\text{left}$ 和 $\text{right}$，分别得到最后一只向左移动的蚂蚁从木板上掉下来的时刻和最后一只向右移动的蚂蚁从木板上掉下来的时刻，其中的最大时刻就是最后一只蚂蚁从木板上掉下来的时刻。

代码

class Solution {
    public int getLastMoment(int n, int[] left, int[] right) {
        int leftLast = 0, rightLast = 0;
        for (int ant : left) {
            leftLast = Math.max(leftLast, ant);
        }
        for (int ant : right) {
            rightLast = Math.max(rightLast, n - ant);
        }
        return Math.max(leftLast, rightLast);
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是木板的长度。需要遍历数组 $\text{left}$ 和 $\text{right}$ 各一次，由于 $\text{left}$ 和 $\text{right}$ 的元素各不相同，因此两个数组的长度之和最大为 $n+1$，对于两个数组中的每个元素，计算从木板上掉下来的时刻的时间是常数，因此总时间复杂度是 $O(n)$。

• 空间复杂度：$O(1)$。