LintCode 814. Shortest Path in Undirected Graph

最新推荐文章于 2020-09-15 17:01:11 发布

原创最新推荐文章于 2020-09-15 17:01:11 发布

· 151 阅读

0 ·

版权

LintCode题解专栏收录该内容

24 篇文章

订阅专栏

本文深入讲解了图中BFS（宽度优先搜索）算法的应用，特别是如何通过分层BFS计算两个节点之间的最短路径。文章提供了详细的代码示例，展示了如何使用队列和哈希表来实现这一过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这道图是图里BFS的经典题！算最短路，其实就是分层BFS（分层BFS的关键就是每次for一下queue的大小，一层一层的pop出元素，便可以用一个变量记录层深）

亦可以用一个hash表，来记录distance。

/**
 * Definition for Undirected graph.
 * struct UndirectedGraphNode {
 *     int label;
 *     vector<UndirectedGraphNode *> neighbors;
 *     UndirectedGraphNode(int x) : label(x) {};
 * };
 */
class Solution {
public:
    /**
     * @param graph: a list of Undirected graph node
     * @param A: nodeA
     * @param B: nodeB
     * @return:  the length of the shortest path
     */
    int shortestPath(vector<UndirectedGraphNode*> graph, UndirectedGraphNode* A, UndirectedGraphNode* B) {
        queue<UndirectedGraphNode *> q;
        q.push(A);
        
        int distance = 0;
        
        while (!q.empty()) {
            int size = q.size();
            for (int i = 0; i < size; i++) {
                UndirectedGraphNode * head = q.front();
                q.pop();
                if (head == B) {
                    return distance;
                }
                for (UndirectedGraphNode * neighbor: head->neighbors) {
                    q.push(neighbor);
                }
            }
            distance++;
        }
        return -1;
    }
};