Looks like graph algorithms comes in handy at the end