如何获取bfs java中两个顶点之间的所有边

时间:2016-11-15 05:59:35

标签: java data-structures graph breadth-first-search bigdata

目前我在两个顶点之间只有一对一的关系。我只想处理两个顶点之间的多个关系。我怎么能这样做?

我目前的代码是:

 public Collection<Vertex<V, E>> bfs() {

    Queue<Graph.Vertex<V, E>> queue = new ArrayBlockingQueue<>(this.getVertices().size());
    Collection<Vertex<V, E>> queryVertices = new LinkedList<>();
    Vertex<V, E> source = this.vertices.get(0);
    Set<Vertex<V, E>> visited = new HashSet<>();

    visited.add(source);
    queue.add(source);
    queryVertices.add(source);

    while (!queue.isEmpty()) {
        Graph.Vertex<V, E> v = queue.remove();
        Graph.Vertex<V, E> w;
        while ((w = getAdjUnvisitedVertex(v, visited)) != null) {
            visited.add(w);
            queue.add(w);
            queryVertices.add(w);
        }
    }

    return queryVertices;
}

private Vertex<V, E> getAdjUnvisitedVertex(Vertex<V, E> v, Set<Vertex<V, E>> visited) {

    for (Graph.Edge<V, E> edge : v.edges) {
        if (!visited.contains(edge.getTo())) {
            return edge.getTo();
        }
    }
    return null;
}

1 个答案:

答案 0 :(得分:0)

在Vertex类中尝试添加相应节点的HashMap或ArrayList。在添加边时,将ID(或Vertex的其他一些识别功能)添加到ArrayList / HashMap中,然后每次要查看每个Vertex所连接的节点时,只需检查ArrayList / HashMap即可。这也将帮助您实现有效的addEdge算法,该算法不会添加节点之间已存在的边。还要记住,如果您正在实现有向图节点A - &gt;节点B!=节点B - &gt;节点A,因此请确保不要在实现中留下边缘。