1503. 所有蚂蚁掉下来前的最后一刻
Difficulty: 中等
有一块木板,长度为 n
个 单位 。一些蚂蚁在木板上移动,每只蚂蚁都以 每秒一个单位 的速度移动。其中,一部分蚂蚁向 左 移动,其他蚂蚁向 右 移动。
当两只向 不同 方向移动的蚂蚁在某个点相遇时,它们会同时改变移动方向并继续移动。假设更改方向不会花费任何额外时间。
而当蚂蚁在某一时刻 t
到达木板的一端时,它立即从木板上掉下来。
给你一个整数 n
和两个整数数组 left
以及 right
。两个数组分别标识向左或者向右移动的蚂蚁在 t = 0
时的位置。请你返回最后一只蚂蚁从木板上掉下来的时刻。
示例 1:
输入:n = 4, left = [4,3], right = [0,1]
输出:4
解释:如上图所示:
-下标 0 处的蚂蚁命名为 A 并向右移动。
-下标 1 处的蚂蚁命名为 B 并向右移动。
-下标 3 处的蚂蚁命名为 C 并向左移动。
-下标 4 处的蚂蚁命名为 D 并向左移动。
请注意,蚂蚁在木板上的最后时刻是 t = 4 秒,之后蚂蚁立即从木板上掉下来。(也就是说在 t = 4.0000000001 时,木板上没有蚂蚁)。
示例 2:
输入:n = 7, left = [], right = [0,1,2,3,4,5,6,7]
输出:7
解释:所有蚂蚁都向右移动,下标为 0 的蚂蚁需要 7 秒才能从木板上掉落。
示例 3:
输入:n = 7, left = [0,1,2,3,4,5,6,7], right = []
输出:7
解释:所有蚂蚁都向左移动,下标为 7 的蚂蚁需要 7 秒才能从木板上掉落。
示例 4:
输入:n = 9, left = [5], right = [4]
输出:5
解释:t = 1 秒时,两只蚂蚁将回到初始位置,但移动方向与之前相反。
示例 5:
输入:n = 6, left = [6], right = [0]
输出:6
提示:
1 <= n <= 10^4
0 <= left.length <= n + 1
0 <= left[i] <= n
0 <= right.length <= n + 1
0 <= right[i] <= n
1 <= left.length + right.length <= n + 1
left
和right
中的所有值都是唯一的,并且每个值 只能出现在二者之一 中。
Solution
两只蚂蚁相遇后会立即掉头,可以等价为两只蚂蚁互相穿过。于是各蚂蚁间便毫无关系,只需要找到最后掉下来的那只蚂蚁。
class Solution(object):
def getLastMoment(self, n, left, right):
"""
:type n: int
:type left: List[int]
:type right: List[int]
:rtype: int
"""
return max(left+[n-x for x in right])
看见一个类似题, P1007 独木桥
又看见一个类似题 蚂蚁感冒 , 原文解释得不清楚但是代码对了((
#include <iostream>
#include <cmath>
using namespace std;
int main(){
int n;
cin >> n;
int pivot, left = 0, right = 0;
cin >> pivot;
int x;
for(int i = 1; i < n; i++){
cin >> x;
if(abs(x) < abs(pivot) && x > 0) right++;
if(abs(x) > abs(pivot) && x < 0) left++;
}
//他这里判断了在感冒源蚂蚁下杆过程中有没有跟它对头的
//没有的话,当然谁都没传染上,因为背后反向和身前同向的都追不上
//有的话,这些对头蚂蚁可以把感冒传给感冒源蚂蚁背后和自己对头的蚂蚁
if((pivot < 0 && right == 0)|| pivot > 0 && left == 0){
cout << "1";
}else{
cout << left + right + 1;
}
return 0;
}