I used sets to store distances and for every 2nd query i traverse to kth element in the set and print it.
How to get rid of TLE?
Just store distances in heap.
Whenever you are asked to get kth distanced hostel, just remove k-1 elements from heap and then Give the answer.
Please have a look at this very short code, its very easy to understand.