GraphBFSSingleSourcePathOptimize.java

/***********************************************************
 * @Description : 单源路径问题优化：BFS遍历到target就提前退出，这样可以极大地节省递归的成本,当前图的构造就只是为了求解当前两个点的路径
 * @author      : 梁山广(Laing Shan Guang)
 * @date        : 2019-08-09 16:37
 * @email       : liangshanguang2@gmail.com
 ***********************************************************/
package Chapter05BreadthFirstTraversal.Section4GraphBFSSingleSourcePathOptimize;

import Chapter02GraphExpress.Graph;

import java.util.*;

public class GraphBFSSingleSourcePathOptimize {

    private Graph graph;
    /**
     * 顶点的访问情况的数组
     */
    private boolean[] visited;

    /**
     * 广度优先遍历的顺序结果(只有一种，不想DFS有前序、后序两种)
     */
    private List<Integer> orderList = new ArrayList<>();

    /**
     * 单源路径的起点、目标点
     */
    private int source, target;

    private int[] pre;

    public GraphBFSSingleSourcePathOptimize(Graph graph, int source,int target) {
        graph.validateVertex(source);
        graph.validateVertex(target);
        this.graph = graph;
        this.source = source;
        this.target = target;
        this.visited = new boolean[graph.V()];
        // 给存储单源路径的数组赋值
        pre = new int[graph.V()];
        Arrays.fill(pre, -1);
        // 因为单源路径问题是和连通分量无关的，所以dfs()要用最早没有考虑连通分量的那版
        bfs();
    }

    /**
     * 从source点开始进行广度优先遍历
     */
    private void bfs() {
        Queue<Integer> queue = new ArrayDeque<>();
        queue.add(source);
        visited[source] = true;
        pre[source] = source;

        if (source == target){
            return;
        }
        while (!queue.isEmpty()) {
            int v = queue.remove();
            orderList.add(v);
            for (int w : graph.adj(v)) {
                if (!visited[w]) {
                    queue.add(w);
                    visited[w] = true;
                    pre[w] = v;
                    if (w == target){
                        return;
                    }
                }
            }
        }
    }

    /**
     * 判断图的遍历起点是否和target点连通，实际只需要看下visit[v]是否为true即可，为true表示在一个连通分量上，肯定是连通地
     *
     * @return 判断图的遍历起点是否和target点连通
     */
    public boolean isConnected() {
        return visited[target];
    }

    public boolean[] getVisited() {
        return visited;
    }

    /**
     * 找到起点source到目标定点target的路径
     *
     * @return 可迭代的对象，一般是集合，用于存储source到target的完整路径
     */
    public Iterable<Integer> path() {
        List<Integer> pathList = new ArrayList<>();
        // source到target有路径才进行路径查找
        if (isConnected()) {
            // 用pre数组从target一直找到source点，记录下中间经过的所有点，就是要求的单源路径
            int current = target;
            while (current != source) {
                pathList.add(current);
                current = pre[current];
            }
            // 起点要加上
            pathList.add(source);
            // 因为是从source到target的路径，所以要颠倒下
            Collections.reverse(pathList);
            return pathList;
        } else {
            // 没有路径就直接返回空集合
            return pathList;
        }
    }

    public Iterable<Integer> getOrderList() {
        return orderList;
    }
}