Учитывая этот график:
Как я могу получить все подключенные вершины из vertexID?
Например, из VertexId 5 он должен вернуть 5-3-7-8-10.
CollectNeighbors возвращает только первые соседние.
Я пытаюсь использовать pregel, но не знаю, как начать с определенной вершины. Я не хочу вычислять все узлы.
Спасибо!
Я только что заметил, что график направлен. то вы можете использовать код примера кратчайшего пути здесь . если расстояние до определенного узла не равно бесконечности, вы можете добраться до этого узла.
или есть лучшая идея, вы можете изменить алгоритм кратчайшего пути, чтобы удовлетворить ваши потребности.
import org.apache.spark.graphx.{Graph, VertexId}
import org.apache.spark.graphx.util.GraphGenerators
// A graph with edge attributes containing distances
val graph: Graph[Long, Double] =
GraphGenerators.logNormalGraph(sc, numVertices = 100).mapEdges(e => e.attr.toDouble)
val sourceId: VertexId = 42 // The ultimate source
// Initialize the graph such that all vertices except the root have canReach = false.
val initialGraph: Graph[Boolean, Double] = graph.mapVertices((id, _) => id == sourceId)
val sssp = initialGraph.pregel(false)(
(id, canReach, newCanReach) => canReach || newCanReach, // Vertex Program
triplet => { // Send Message
if (triplet.srcAttr && !triplet.dstAttr) {
Iterator((triplet.dstId, true))
} else {
Iterator.empty
}
},
(a, b) => a || b // Merge Message
)
println(sssp.vertices.collect.mkString("\n"))
