Advertisement
Guest User

Untitled

a guest
Feb 24th, 2020
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.95 KB | None | 0 0
  1. trains = {}
  2.  
  3. while True:
  4.     command = input()
  5.     if command == "Slide rule":
  6.         break
  7.     tokens = command.split(":")
  8.     town = tokens[0]
  9.     tokens = tokens[1].split("->")
  10.     passengers = int(tokens[1])
  11.     if tokens[0].isdigit():
  12.         time = int(tokens[0])
  13.         if town not in trains:
  14.             trains[town] = [time, passengers]
  15.         else:
  16.             if trains[town][0] >= time:
  17.                 trains[town][0] = time
  18.             elif trains[town][0] == 0:
  19.                 trains[town][0] = time
  20.             trains[town][1] += passengers
  21.     else:
  22.         if town in trains:
  23.             trains[town][0] = 0
  24.             if trains[town][1] >= passengers:
  25.                 trains[town][1] -= passengers
  26.  
  27.  
  28.  
  29. sorted_trains = sorted(trains.items(), key= lambda x: (x[1][0], x))
  30.  
  31. for (key, value) in sorted_trains:
  32.     if value[0] > 0 and value[1] > 0:
  33.         print(f"{key} -> Time: {value[0]} -> Passengers: {value[1]}")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement