iframe-proxy | Sunbelt Computer Software

History

148 lines (125 loc) · 4.47 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

// @saorav21994

// TC : O(1) - asymptotically

// SC : O(station^2) - for each station -> other station we need to know the details

class UndergroundSystem {

HashMap<Integer, Pair<String, Integer>> srcStation;

HashMap<String, station> stationDetail;

public UndergroundSystem() {

srcStation = new HashMap<>();

stationDetail = new HashMap<>();

// System.out.println("init");

}

public void checkIn(int id, String stationName, int t) {

srcStation.put(id, new Pair<>(stationName, t));

// System.out.println("checkIn");

}

public void checkOut(int id, String stationName, int t) {

Pair<String, Integer> pair = srcStation.get(id);

String dst = stationName;

String src = pair.getKey();

srcStation.remove(id);

int timeTaken = t-pair.getValue();

HashMap<String, Pair<Integer, Integer>> tmp;

Pair<Integer, Integer> timecount;

if (stationDetail.containsKey(src) == true) {

tmp = stationDetail.get(src).time;

if (tmp.containsKey(dst) == true) {

timecount = tmp.get(dst);

}

else {

timecount = new Pair<>(0, 0);

}

int ptotal = timecount.getKey();

int pcount = timecount.getValue();

pcount += 1;

ptotal += timeTaken;

timecount = new Pair<>(ptotal, pcount);

}

else {

timecount = new Pair<>(timeTaken, 1);

tmp = new HashMap<>();

}

tmp.put(dst, timecount);

stationDetail.put(src, new station(tmp));

// System.out.println("checkOut");

}

public double getAverageTime(String startStation, String endStation) {

HashMap<String, Pair<Integer, Integer>> tmp;

Pair<Integer, Integer> timecount;

station stat = stationDetail.get(startStation);

tmp = stat.time;

timecount = tmp.get(endStation);

double res = (double)((double)timecount.getKey()/(double)timecount.getValue());

// System.out.println("res = " + res);

return res;

}

public class station {

private HashMap<String, Pair<Integer, Integer>> time = new HashMap<>();

public station(HashMap<String, Pair<Integer, Integer>> time) {

this.time = time;

}

/**

* Your UndergroundSystem object will be instantiated and called as such:

* UndergroundSystem obj = new UndergroundSystem();

* obj.checkIn(id,stationName,t);

* obj.checkOut(id,stationName,t);

* double param_3 = obj.getAverageTime(startStation,endStation);

// Author: @romitdutta10

// TC: O(1)

// SC: O(1)

// Problem: https://leetcode.com/problems/design-underground-system/

class UndergroundSystem {

Map<Integer, Station> stations;

Map<String, Time> averageTimes;

public UndergroundSystem() {

stations = new HashMap<>();

averageTimes = new HashMap<>();

}

public void checkIn(int id, String stationName, int t) {

stations.put(id, new Station(stationName, t));

}

public void checkOut(int id, String stationName, int t) {

Station inPoint = stations.get(id);

int travelTime = t - inPoint.time;

String key = inPoint.stationName + " " + stationName;

Time time = averageTimes.getOrDefault(key, new Time());

time.total += travelTime;

time.noOfPassegers++;

averageTimes.put(key, time);

}

public double getAverageTime(String startStation, String endStation) {

String key = startStation + " " + endStation;

Time time = averageTimes.get(key);

return time.total * 1.0 / time.noOfPassegers;

}

class Station {

String stationName;

int time;

Station(String stationName, int time) {

this.stationName = stationName;

this.time = time;

}

class Time {

int total;

int noOfPassegers;

Time() {

this.total = 0;

this.noOfPassegers = 0;

}

Time(int total, int noOfPassegers) {

this.noOfPassegers = noOfPassegers;

this.total = total;

}

/**

* Your UndergroundSystem object will be instantiated and called as such:

* UndergroundSystem obj = new UndergroundSystem();

* obj.checkIn(id,stationName,t);

* obj.checkOut(id,stationName,t);

* double param_3 = obj.getAverageTime(startStation,endStation);

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

DesignUndergroundSystem.java

Latest commit

History

DesignUndergroundSystem.java

File metadata and controls