LeetCode每日一题: 853. 车队

853. 车队

标签: 贪心算法

N 辆车沿着一条车道驶向位于 target 英里之外的共同目的地。
每辆车 i 以恒定的速度 speed[i] （英里/小时），从初始位置 position[i] （英里） 沿车道驶向目的地。
一辆车永远不会超过前面的另一辆车，但它可以追上去，并与前车以相同的速度紧接着行驶。
此时，我们会忽略这两辆车之间的距离，也就是说，它们被假定处于相同的位置。
车队 是一些由行驶在相同位置、具有相同速度的车组成的非空集合。注意，一辆车也可以是一个车队。
即便一辆车在目的地才赶上了一个车队，它们仍然会被视作是同一个车队。
会有多少车队到达目的地?

思路

先过滤出有用的信息并处理, 观察题目后我们可以知道:

• 问题: 会有多少车队到达目的地?

• 限制条件: 一辆车永远不会超过前面的另一辆车, 即初始位置(position)是结果的主要影响因素

• 速度是另一个影响因素

• 从已知条件出发:

  1. 先对根据初始位置对每个车的信息进行排序
  2. 计算最接近目的地的一辆车的到达时间, 与它后方的一辆车进行比较, 如果后一辆车的到达时间<= 当前车的到达时间, 则因为限制条件的存在, 它们会同时到达
  3. 反之, 则它们不同时到达, 后一辆车形成新的车队, 为同一车队
  4. 边界条件: 当N=0时, 会有0个车队到达; N = 1时, 1个车队到达

解法

class Solution:
    def carFleet(self, target: int, position: List[int], speed: List[int]) -> int:
        lst = sorted(zip(position, speed))
        times = [(target-p)/s for p, s in lst ]
        ans = 0
        
        while len(times) > 1:
            lead = times.pop()
            if lead < times[-1]:
                ans += 1
            else:
                times[-1] = lead
        # 当times = 1时,不进入循环, 最后一队车没有可比较的对象, 没有+1, 所以此处需要+1
        # 当time = 0时, 不进入循环, 没有车队, 所以此处为0
        # 合并上述两项, 所以采用如下写法
        return ans + bool(times)