BFS及其应用

内容概要:

  1. BFS类的实现
  2. BFS求解连通分量
  3. BFS求解无向图点对之间的一条最短路径
  4. BFS判定无环图和二分图
  5. BFS与DFS的联系

树与图的广度优先遍历对比

树的层次遍历其实就是一种广度优先遍历。图的广度优先遍历和树的广度优先遍历非常类似,区别只在于图要对节点是否被遍历做记录。

广度优先遍历类

BFS只是和DFS的遍历逻辑不通,有了DFS的基础,不难给出BFS的相关类。
广度优先遍历到某个顶点后,要遍历与它相邻的所有顶点。这个过程使用队列实现。

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

public class GraphBFS {
    private Graph G;
    private boolean[] visited;
    private ArrayList<Integer> order = new ArrayList<>();

    public GraphBFS(Graph G){
        this.G = G;
        visited = new boolean[G.V()];
        for(int v = 0; v < G.V(); v ++)
            if(!visited[v])
                bfs(v);
    }

    private void bfs(int s){
        // 从 s 开始进行BFS
        Queue<Integer> queue = new LinkedList<>();
        queue.add(s);
        visited[s] = true;
        while(!queue.isEmpty()) {
            int v = queue.remove(); // 取队头
            order.add(v);

            for (int w : G.adj(v))
                if (!visited[w]) {
                    queue.add(w);
                    visited[w] = true;
                }
        }
    }
    public Iterable<Integer> order(){
        return order;
    }
    public static void main(String[] args){
        Graph g = new Graph("g.txt");
        GraphBFS graphBFS = new GraphBFS(g);
        System.out.println("BFS :" + graphBFS.order());
    }
}

广度优先遍历一样可以解决求图的连通分量个数、环检测、二分图检测等。

BFS解决单源路径问题

BFS的性质
遍历时由顶点到顶点一定是跨越边数最少的路径。
求解路径问题
两个顶点在同一连通分量表示有路径,从一点出发,记录经过的顶点,直到到达另一顶点这就是一条路径。具体做法就是记录每个顶点是从哪里来的,由于会有访问标记,所以这个值是唯一的,最后从终点到源点反推即得到路径。
由BFS的性质,这个路径一定是跨过的边数最少的路径,如果是无权图,就是最短路径。

import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.Queue;

public class ShortestPath {
    private Graph G;
    private int s;
    private boolean[] visited;
    private int[] pre;
    private int[] dis;

    public ShortestPath(Graph G, int s){
        this.G = G;
        this.s = s;
        visited = new boolean[G.V()];
        dis = new int[G.V()];
        pre = new int[G.V()];

        for(int i = 0; i < G.V(); i ++) {
            pre[i] = -1;
            dis[i] = -1;
        }

        bfs(s);
    }

    private void bfs(int s){
        // 从 s 开始进行BFS
        Queue<Integer> queue = new LinkedList<>();
        queue.add(s);
        visited[s] = true;
        pre[s] = s;
        dis[s] = 0;

        while(!queue.isEmpty()) {
            int v = queue.remove(); // 取队头

            for (int w : G.adj(v))
                if (!visited[w]) {
                    queue.add(w);
                    visited[w] = true;
                    pre[w] = v;
                    dis[w] = dis[v] + 1;// 距离加一
                }
        }
    }

    public boolean isConnectedTo(int t){
        G.validateVertex(t);
        return visited[t];// 意味着 s 与 t 在同一连通分量;
    }

    public Iterable<Integer> path(int t){
        // s -> t 的路径

        ArrayList<Integer> res = new ArrayList<>();
        if(!isConnectedTo(t)) return res;
        int cur = t;
        while(cur != s){
            res.add(cur);
            cur = pre[cur];
        }
        res.add(s);
        Collections.reverse(res);
        return res;
    }
    public int dis(int t){
        // s -> t 的距离,不关注路径,只关注最短距离
        G.validateVertex(t);
        return dis[t];

    }
    public static void main(String[] args){
        Graph g = new Graph("g.txt");
        ShortestPath sp = new ShortestPath(g,0);
        System.out.println("0 -> 3: " + sp.dis(3));
    }
}

由上也不难得到所有点对之间的最短路径。

BFS求解连通分量

与DFS解决连通分量问题思路一样,将visited数组定义为整型数组,初值为-1表示没有访问过这个顶点,而非负值就代表顶点所属的连通分量。

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

public class ConnectedComponents {
    private Graph G;
    private int[] visited;
    private int cccount = 0;

    public ConnectedComponents(Graph G){
        this.G = G;
        visited = new int[G.V()];
        for(int i = 0; i < visited.length; i ++)
            visited[i] = -1;

        for(int v = 0; v < G.V(); v ++)
            if(visited[v] == -1) {
                bfs(v, cccount);
                cccount ++;
            }
    }
    private void bfs(int s, int ccid){
        Queue<Integer> queue = new LinkedList<>();
        queue.add(s);
        visited[s] = ccid;

        while(!queue.isEmpty()){
            int v = queue.remove();

            for(int w: G.adj(v))
                if(visited[w] == -1){
                    queue.add(w);
                    visited[w] = ccid;
                }
        }
    }

    public int count(){ // 返回连通分量个数
        return cccount;
    }

    public boolean isConnected(int v, int w){
        return visited[v] == visited[w];
    }

    public ArrayList<Integer>[] components(){
        // 返回各个连通分量
        ArrayList<Integer>[] res = new ArrayList[cccount];
        for(int i = 0; i < cccount; i ++)
            res[i] = new ArrayList<>();
        for(int v = 0; v < G.V(); v ++)
            res[visited[v]].add(v);
        return res;
    }
    public static void main(String args[]){
        Graph g = new Graph("g.txt");
        ConnectedComponents cc = new ConnectedComponents(g);
        ArrayList<Integer>[] comp = cc.components();
        for(int ccid = 0; ccid < comp.length; ccid ++) {
            System.out.print(ccid + ": ");
            for (int w : comp[ccid])
                System.out.print(w + " ");
            System.out.println();
        }
    }
}

BFS求解环检测

与DFS解决该问题思路一样,如果一个被访问的顶点再次被访问到,且这个顶点不是它前一个节点,则说明图中存在环。

import java.util.LinkedList;
import java.util.Queue;

public class CycleDetection {
    private Graph G;
    private boolean[] visited;
    private int[] pre;
    private boolean hasCycle = false;

    public CycleDetection(Graph G){
        this.G = G;
        visited = new boolean[G.V()];
        pre = new int[G.V()];
        for(int v = 0; v < G.V(); v ++)
            pre[v] = -1;

        for(int v = 0; v < G.V(); v ++)
            if(!visited[v])
                if(bfs(v)){ // 如果通过BFS检测出环
                    hasCycle = true;
                    break;
                }
    }
    private boolean bfs(int s){
        Queue<Integer> queue = new LinkedList<>();
        queue.add(s);
        visited[s] = true;
        pre[s] = s;
        while(!queue.isEmpty()){
            int v = queue.remove();
            for(int w: G.adj(v)){
                if(!visited[w]){
                    queue.add(w);
                    visited[w] = true;
                    pre[w] = v;
                }
                // 这里是逻辑的关键
                // 如果 w 已经被访问过了,我们还必须判断,w 不是 v 的上一个节点
                // 如果 w 不是 v 的上一个节点,说明我们找到了一个环
                else if(pre[v] != w)
                    return true;
            }
        }
        return false;
    }
    public boolean hasCycle(){
        return hasCycle;
    }
    public static void main(String[] args) {

        Graph g = new Graph("g.txt");
        CycleDetection cycleDetection = new CycleDetection(g);
        System.out.println(cycleDetection.hasCycle());
    }
}

BFS判定二分图

通过染色操作完成,用两种颜色对图进行染色,如果可以每两个相邻顶点都不同色,则这个图是二分图。

import java.util.LinkedList;
import java.util.Queue;

public class BiPartitionDetection {
    private Graph G;
    private boolean[] visited;
    private int[] colors; //  两种颜色,实际 colors 可以兼有 visited 的功能
    private boolean isBipartite = true;

    public BiPartitionDetection(Graph G){
        this.G = G;
        visited = new boolean[G.V()];
        colors = new int[G.V()];

        for(int v = 0; v < G.V(); v ++)
            colors[v] = -1;

        for(int v = 0; v < G.V(); v ++)
            if(!visited[v])
                if(!bfs(v)){
                    isBipartite = false;
                    break;
                }
    }

    private boolean bfs(int s){
        // 从s 点开始看整张图是否是二分图
        Queue<Integer> queue = new LinkedList<>();
        queue.add(s);
        visited[s] = true;
        colors[s] = 0;
        while (!queue.isEmpty()){
            int v = queue.remove();
            for(int w: G.adj(v)) {
                if (!visited[w]) {
                    queue.add(w);
                    visited[w] = true;
                    colors[w] = 1 - colors[v];
                }
                else if(colors[v] == colors[w]){
                    return false;
                }
            }
        }
        return true;
    }

    public static void main(String[] args) {

        Graph g = new Graph("g.txt");
        BiPartitionDetection bipartitionDetection = new BiPartitionDetection(g);
        System.out.println(bipartitionDetection.isBipartite);
    }
}

BFS与DFS之间的联系

观察BFS和DFS的非递归代码:
BFS

visited[0...V-1] = false;
for(int v = 0; v < G.V(); v ++)
    if(!visited[v])
        bfs(v);

void bfs(int s){
    Queue<Integer> queue = new LinkedList<>();
    queue.add(s);
    visited[s] = true;
    while(!queue.isEmpty()) {
        int v = queue.remove(); 
        order.add(v);
        for (int w : G.adj(v))
            if (!visited[w]) {
                queue.add(w);
                visited[w] = true;
            }
    }
}

DFS

visited[0...V-1] = false;
for(int v = 0; v < G.V(); v ++)
    if(!visited[v])
        bfs(v);

void dfs(int s){
    Stack<Integer> stack = new Stack<>();
    stack.add(s);
    visited[s] = true;
    while(!stack.isEmpty()) {
        int v = stack.remove(); 
        order.add(v);
        for (int w : G.adj(v))
            if (!visited[w]) {
                stack.add(w);
                visited[w] = true;
            }
    }
}

它们的非递归代码的区别只在于数据结构使用的不同,一个用栈一个用队列,而其它的逻辑完全一样。也就只是在容器中循环取元素的顺序不同而已。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 220,458评论 6 513
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 94,030评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 166,879评论 0 358
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,278评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,296评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 52,019评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,633评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,541评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,068评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,181评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,318评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,991评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,670评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,183评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,302评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,655评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,327评论 2 358

推荐阅读更多精彩内容