Trakai's bus station is a depressing affair with little in the way of facilities. To use the toilets you need to get a key from the ticket window, which isn't always open. Buy tickets from the driver for the journey back to Vilnius.
