今天第一次参加周赛,作为稳稳的前三选手,看了一下第三题的设计,有个Java题解做的比较简洁,我就照着实现了一遍,下面是题目:
力扣5370
这个题解是抄的评论区的一个题解,里面还有一点疑问,但是力扣的账号不绑手机不让发评论,不然我就问问他了,手机号被我有一次新建了一个账号,吐了。
import java.util.*;
class UndergroundSystem {
//Key值是站点的名称,station保存了该站点的全部出入信息
HashMap<String,Station> stationMap;
public UndergroundSystem() {
stationMap = new HashMap<>();
}
public void checkIn(int id, String stationName, int t) {
Station station = stationMap.getOrDefault(stationName,new Station());
station.in.put(id,t);
stationMap.put(stationName,station);
}
public void checkOut(int id, String stationName, int t) {
Station station = stationMap.getOrDefault(stationName,new Station());
station.out.put(id,t);
stationMap.put(stationName,station);
}
/**
到这里我就有些疑问了,如果按照这种解法,如果有一个人两次进行了同样的操作,比如
undergroundSystem.checkIn(45, "Leyton", 3);
undergroundSystem.checkIn(45, "Waterloo", 8);
undergroundSystem.checkIn(45, "Leyton", 10);
undergroundSystem.checkOut(45, "Waterloo", 15);
undergroundSystem.getAverageTime("Paradise", "Cambridge");
这个测试样例,在checkin和checkout函数中,后面一次的乘车记录就完全覆盖掉了前面
这次的乘车记录,于是我就用这个测试了一下。然后我在测试的途中发现,即使覆盖掉了一
此乘车记录也对两个站点的平均乘车时间毫无影响,因为两个站点的乘车时间是相等恒定的,不愧是大佬QAQ
**/
public double getAverageTime(String startStation, String endStation) {
double nums = 0;
double sum = 0;
double average = 0;
Station start = stationMap.get(startStation);
Station end = stationMap.get(endStation);
for(Integer pass:start.in.keySet()){
if(end.out.containsKey(pass)){
sum += end.out.get(pass) - start.in.get(pass);
nums++;
}
}
average = sum/nums;
return average;
}
private class Station{
//保存一个站点的乘客出入信息,包括乘客id和出入时间
HashMap<Integer,Integer> in = new HashMap<Integer,Integer>();
HashMap<Integer,Integer> out = new HashMap<Integer,Integer>();
}
}