From fd2c57b8a4741fab8c07d1b9888dd4f5ba44cf1e Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Mon, 1 Dec 2014 15:32:57 -0800 Subject: [PATCH] Update engine_robust.cc --- src/engine_robust.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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