树与图的宽度优先遍历

137 阅读2分钟

「这是我参与2022首次更文挑战的第22天,活动详情查看:2022首次更文挑战

树与图的宽度优先遍历本质上考察的还是宽度优先遍历的框架。

宽度优先遍历框架

把首元素入队列,当队列不为空时,拿出队头t,扩展t的所有邻点x,if(x未遍历) 那么x入队列,并且更新x点的距离。我们可以依靠这个框架对宽度优先遍历的题目进行思考。

转化为代码:

// 代码模板
queue<int> q;
st[1] = true; // 表示1号点已经被遍历过
q.push(1);
​
while (q.size())
{
    int t = q.front();
    q.pop();
​
    for (int i = h[t]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (!st[j])
        {
            st[j] = true; // 表示点j已经被遍历过
            q.push(j);
        }
    }
}

大家一起来看下这个模板题

模板题

给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环。

所有边的长度都是 1,点的编号为 1∼n。

请你求出 1 号点到 n 号点的最短距离,如果从 1 号点无法走到 n 号点,输出 −1。

输入格式

第一行包含两个整数 n 和 m。

接下来 m 行,每行包含两个整数 a 和 b,表示存在一条从 a 走到 b 的长度为 1 的边。

输出格式

输出一个整数,表示 1 号点到 n 号点的最短距离。

数据范围

1≤n,m≤105

输入样例:

4 5
1 2
2 3
3 4
1 3
1 4

输出样例:

1

思路:

我们把点与点的关系用邻接表进行表示,由于这是一张无向图,所以只要在两点之间建立一条边即可。我们把第一个起始点放入队头中,再用d数组标识这个点是否被遍历过,没有被遍历过的元素都会进入队列,只要队列中还存有元素时我们就会继续遍历。

代码:

#include <iostream>
#include <algorithm>
#include <cstring>
​
using namespace std;
​
const int N = 100010;
int h[N], e[N], ne[N], idx;
int q[N],d[N];
int n, m;
void add(int a, int b)
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++;
}
int bfs()
{
    int hh = 0, tt = 0;
    q[0] = 1;// 队列从第一个点开始
    memset(d, -1, sizeof d);
    d[1] = 0;// 表示第一个点被遍历过了
    while(hh <= tt)
    {
        int t = q[hh ++];
        for(int i = h[t];i != -1;i = ne[i])
        {
            int g = e[i];
            if(d[g] == -1)
            {
                d[g] = d[t] + 1;
                q[++ tt] = g;
            }
        }
    }
    return d[n];
}
int main()
{
    cin >> n >> m;
    memset(h, -1, sizeof h);
    for(int i = 0;i < m;i ++)
    {
        int a, b;
        cin >> a >> b;
        add(a, b);
    }
    cout << bfs() << endl;
    return 0;
}