使用pregel graphx激发一对一的最短路径

时间:2017-08-09 15:44:06

标签: scala apache-spark shortest-path spark-graphx

我尝试使用link

中的代码找到从单个源到n个顶点的最短路径
val graph: Graph[Long, Double] =
  GraphGenerators.logNormalGraph(sc, numVertices = 100).mapEdges(e => e.attr.toDouble)
val sourceId: VertexId = 42

val initialGraph = graph.mapVertices((id, _) =>
    if (id == sourceId) 0.0 else Double.PositiveInfinity)
val sssp = initialGraph.pregel(Double.PositiveInfinity)(
  (id, dist, newDist) => math.min(dist, newDist),
  triplet => {
    if (triplet.srcAttr + triplet.attr < triplet.dstAttr) {
      Iterator((triplet.dstId, triplet.srcAttr + triplet.attr))
    } else {
      Iterator.empty
    }
  },
  (a, b) => math.min(a, b)
)
println(sssp.vertices.collect.mkString("\n"))

它给出了从42到N个顶点的输出最短路径。 但是,如何找到单一来源到单一目的地之间的最短路径? 即source = 42,dest = 135然后我想找到它们之间的最短路径。

感谢

1 个答案:

答案 0 :(得分:0)

否则,您将无法找到两个顶点之间的最短距离,即,不浏览整个图形。因为即使目的地收到您的消息后,即使您停止探索,也无法保证它是最短的距离。 因此,该算法在复杂度方面很好。您应该只从获得的不同目的地中选择sssp值。