diff --git a/src/engine_robust.cc b/src/engine_robust.cc index 44497bfbe..be950aa98 100644 --- a/src/engine_robust.cc +++ b/src/engine_robust.cc @@ -303,7 +303,8 @@ ShortestDist(const std::pair &node_value, res = dist_in[i].first; size = dist_in[i].second; } } - return std::make_pair(res, size); + // add one hop + return std::make_pair(res + 1, size); } /*! * \brief message passing function, used to decide the