Could you have just used dijkstra or breadth first search?