1396. 设计地铁系统-CSDN博客

checkIn(int id, string stationName, int t)
乘客 ID 为 id，在时间 t 从车站 stationName 进入地铁。一个乘客在任意时刻只会有一个“进站”记录。
checkOut(int id, string stationName, int t)
乘客 ID 为 id，在时间 t 从车站 stationName 离开地铁。
getAverageTime(string startStation, string endStation)
返回所有从 startStation 进入且到达 endStation 离开的乘客的平均出行时间。

要求：

时间 t 是严格递增的，即每位乘客的出站时间一定大于进站时间。
对于每对起止站点，能够多次查询平均时间。
可以假设所有操作调用都是合理的，比如每个乘客的 checkIn 和 checkOut 都成对出现且时序正确。

解题分析

核心需求是实时记录每个乘客的行程开始时间和起点车站，以及统计从一个车站到另一个车站的所有行程总时间和次数，以便计算平均时间。

需要两个核心数据结构：

乘客进站信息映射：
用于记录每个乘客的进站站点和时间，方便在乘客出站时计算行程时长。
行程时间统计映射：
用于累加从某站到另一站的所有行程的总时间及次数，支持计算平均时间。

解题方法

记录进站：
乘客进站时，存储 (id) -> (stationName, t)，表示乘客 ID、进站车站及时间。
计算并更新出站数据：
乘客出站时，根据乘客 ID 查找其进站信息，计算行程时间 t_out - t_in，更新该线路（起点到终点）的累计时间和计数。
获取平均时间：
根据起点和终点站查询累计总时间和次数，计算平均时间返回。

这种设计实现逻辑清晰，且能高效支持多次查询。

代码实现（Python）

class UndergroundSystem:
    def __init__(self):
        # 乘客进站记录：id -> (stationName, time)
        self.check_in_map = {}
        # 线路累计数据：(startStation, endStation) -> [total_time, trip_count]
        self.travel_map = {}

    def checkIn(self, id: int, stationName: str, t: int) -> None:
        self.check_in_map[id] = (stationName, t)

    def checkOut(self, id: int, stationName: str, t: int) -> None:
        start_station, start_time = self.check_in_map.pop(id)
        route_key = (start_station, stationName)
        travel_time = t - start_time
        if route_key not in self.travel_map:
            self.travel_map[route_key] = [0, 0]
        self.travel_map[route_key][0] += travel_time
        self.travel_map[route_key][1] += 1

    def getAverageTime(self, startStation: str, endStation: str) -> float:
        total_time, count = self.travel_map[(startStation, endStation)]
        return total_time / count

示例说明

假设我们有以下操作：

undergroundSystem = UndergroundSystem()
undergroundSystem.checkIn(1, "A", 3)       # 乘客1在时间3从车站A进站
undergroundSystem.checkOut(1, "B", 8)      # 乘客1在时间8从车站B出站，行程时间5
print(undergroundSystem.getAverageTime("A", "B"))  # 输出5.0

undergroundSystem.checkIn(2, "A", 10)
undergroundSystem.checkOut(2, "B", 15)     # 乘客2行程时间5
print(undergroundSystem.getAverageTime("A", "B"))  # 输出(5+5)/2=5.0

undergroundSystem.checkIn(3, "A", 20)
undergroundSystem.checkOut(3, "B", 30)     # 乘客3行程时间10
print(undergroundSystem.getAverageTime("A", "B"))  # 输出(5+5+10)/3=6.67

这展示了如何累计多条路线数据，并返回平均值。