1. 程式人生 > 其它 >Leetcode---3.BFS篇

Leetcode---3.BFS篇

BFS 相對 DFS 的最主要的區別是:BFS 找到的路徑一定是最短的,但代價就是空間複雜度可能比 DFS 大很多。

BFS框架:

// 計算從起點 start 到終點 target 的最近距離
int BFS(Node start, Node target) {
    Queue<Node> q; // 核心資料結構
    Set<Node> visited; // 避免走回頭路

    q.offer(start); // 將起點加入佇列
    visited.add(start);
    int step = 0; // 記錄擴散的步數

    while (q not empty) {
        int sz = q.size();
        /* 將當前佇列中的所有節點向四周擴散 */
        for (int i = 0; i < sz; i++) {
            Node cur = q.poll();
            /* 劃重點:這裡判斷是否到達終點 */
            if (cur is target)
                return step;
            /* 將 cur 的相鄰節點加入佇列 */
            for (Node x : cur.adj()) {
                if (x not in visited) {
                    q.offer(x);
                    visited.add(x);
                }
            }
        }
        /* 劃重點:更新步數在這裡 */
        step++;
    }
}

BFS 的核心資料結構;cur.adj() 泛指 cur 相鄰的節點,比如說二維陣列中,cur 上下左右四面的位置就是相鄰節點;visited 的主要作用是防止走回頭路,大部分時候都是必須的,但是像一般的二叉樹結構,沒有子節點到父節點的指標,不會走回頭路就不需要 visited。