diff --git a/mock_exam/app/src/main/java/org/example/App.java b/mock_exam/app/src/main/java/org/example/App.java index caf7d40..f363a6b 100644 --- a/mock_exam/app/src/main/java/org/example/App.java +++ b/mock_exam/app/src/main/java/org/example/App.java @@ -54,6 +54,14 @@ public class App { this.arrivalTime = arrivalTime; this.stops = stops.clone(); } + + public Station getStartStation() { + return this.stops[0]; + } + + public Station getFinalStation() { + return this.stops[this.stops.length - 1]; + } } class LineData { @@ -183,7 +191,8 @@ public class App { String toStationName = sc.nextLine(); ArrayList trains2 = a.trainsFromTo(fromStationName, toStationName); for (Train t : trains2) { - System.out.println("found " + t.type.toString()); + System.out.println("found " + t.type.toString() + " from " + t.getStartStation().getName() + " to " + + t.getFinalStation().getName()); } sc.close(); }