853. 车队
在一条单行道上,有 n
辆车开往同一目的地。目的地是几英里以外的 target
。
给定两个整数数组 position
和 speed
,长度都是 n
,其中 position[i]
是第 i
辆车的位置, speed[i]
是第 i
辆车的速度(单位是英里/小时)。
一辆车永远不会超过前面的另一辆车,但它可以追上去,并与前车 以相同的速度 紧接着行驶。此时,我们会忽略这两辆车之间的距离,也就是说,它们被假定处于相同的位置。
车队 是一些由行驶在相同位置、具有相同速度的车组成的非空集合。注意,一辆车也可以是一个车队。
即便一辆车在目的地才赶上了一个车队,它们仍然会被视作是同一个车队。
返回到达目的地的 车队数量 。
示例 1:
输入:target = 12, position = [10,8,0,5,3], speed = [2,4,1,1,3] 输出:3 解释: 从 10 和 8 开始的车会组成一个车队,它们在 12 处相遇。 从 0 处开始的车无法追上其它车,所以它自己就是一个车队。 从 5 和 3 开始的车会组成一个车队,它们在 6 处相遇。 请注意,在到达目的地之前没有其它车会遇到这些车队,所以答案是 3。
示例 2:
输入: target = 10, position = [3], speed = [3] 输出: 1 解释: 只有一辆车,因此只有一个车队。
示例 3:
输入: target = 100, position = [0,2,4], speed = [4,2,1] 输出: 1 解释: 以0(速度4)和2(速度2)出发的车辆组成车队,在4点相遇。舰队以2的速度前进。 然后,车队(速度2)和以4(速度1)出发的汽车组成一个车队,在6点相遇。舰队以1的速度前进,直到到达目标。
提示:
n == position.length == speed.length
1 <= n <= 105
0 < target <= 106
0 <= position[i] < target
position
中每个值都 不同0 < speed[i] <= 106
原站题解
golang 解法, 执行用时: 184 ms, 内存消耗: 9.3 MB, 提交时间: 2023-09-19 17:11:50
type node struct { pos int times float64 } func carFleet(target int, position []int, speed []int) int { node := make([]node, len(position)) if len(position) == 0 { return 0 } for i, _ := range position { node[i].pos = position[i] node[i].times = float64(target-position[i]) / float64(speed[i]) } sort.Slice(node, func(i, j int) bool { return node[i].pos > node[j].pos }) sign := node[0].times count := 1 for _, time := range node { if time.times > sign { count++ sign = time.times } } return count }
python3 解法, 执行用时: 316 ms, 内存消耗: 47.7 MB, 提交时间: 2023-09-19 17:10:28
class Solution: def carFleet(self, target: int, position: List[int], speed: List[int]) -> int: res, n = 1, len(speed) ps = sorted(zip(position, speed), key=lambda x:(-x[0], -x[1])) t = (target - ps[0][0]) / ps[0][1] for i in range(1, n): if ps[i][0] + ps[i][1] * t < target: res += 1 t = (target - ps[i][0]) / ps[i][1] return res
cpp 解法, 执行用时: 304 ms, 内存消耗: 94.8 MB, 提交时间: 2023-09-19 17:09:59
class Solution { public: int carFleet(int target, vector<int>& position, vector<int>& speed) { map<int, int> ps; for (int i = 0; i < position.size(); i++) { ps[position[i]] = speed[i]; } stack<float> stk; int ans = 0; for (auto& [pos, spd] : ps) { float time = float(target - pos) / spd; while (!stk.empty() && time >= stk.top()) { stk.pop(); } stk.push(time); } return stk.size(); } };
java 解法, 执行用时: 73 ms, 内存消耗: 55.1 MB, 提交时间: 2023-09-19 17:09:42
class Solution { public int carFleet(int target, int[] position, int[] speed) { int N = position.length; Car[] cars = new Car[N]; for (int i = 0; i < N; ++i) cars[i] = new Car(position[i], (double) (target - position[i]) / speed[i]); Arrays.sort(cars, (a, b) -> Integer.compare(a.position, b.position)); int ans = 0, t = N; while (--t > 0) { if (cars[t].time < cars[t-1].time) ans++; //if cars[t] arrives sooner, it can't be caught else cars[t-1] = cars[t]; //else, cars[t-1] arrives at same time as cars[t] } return ans + (t == 0 ? 1 : 0); //lone car is fleet (if it exists) } } class Car { int position; double time; Car(int p, double t) { position = p; time = t; } }
python3 解法, 执行用时: 232 ms, 内存消耗: 39.6 MB, 提交时间: 2023-09-19 17:09:10
class Solution(object): def carFleet(self, target, position, speed): cars = sorted(zip(position, speed)) times = [float(target - p) / s for p, s in cars] ans = 0 while len(times) > 1: lead = times.pop() if lead < times[-1]: ans += 1 # if lead arrives sooner, it can't be caught else: times[-1] = lead # else, fleet arrives at later time 'lead' return ans + bool(times) # remaining car is fleet (if it exists)