原题链接 : 847. 图中点的层次 - AcWing题库
思路:
用dist数组保存1号节点到各个节点的距离,初始都设为无限大
用st数组标记各个节点有没有走到过。
从1号节点开始,广度优先遍历:
1. 1号节点入队列,dist[1] 的值更新为 0;
2. 如果队列非空,就取出队头,找到队头节点能到的所有节点。如果队头节点
能走到的节点没有标记过,就将节点的dist值更新为队头的dist + 1,然后入
队。
3. 重复步骤2 直到队列为空。
4. 此时,dist数组中就存储了 1 号节点到各个节点的距离了。如果距离是无穷大,则不能到达
,输出距离。
图的存储: 邻接表
* 用 h 数组保存各个节点能到的第一个节点的编号。开始时,尾巴全部为 -1。
* 用 e 数组保存节点的编号, ne数组保存 e 数组对应位置的下一个节点所在的idx
* 用idx作为伪指针。
* 插入边使用 头插法, 例如插入 a -> b。首先把b 节点存入 e数组,e[idx] = b,
然后b节点的后继是h[a], ne[idx] = h[a]. 最后, a的后继更新为b节点的编号,
h[a] = idx, 伪指针指向下一个可以存储节点的位置,idx ++。
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int N = 100010;
int h[N],ne[N],e[N],idx; //邻接表数据结构标准四件套
int dist[N]; // 存储距离
int st[N]; // 标记点是否走到过
int n, m;
void (int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
void bfs()
{
memset(dist, 0x3f, sizeof(dist)); // 初始都没有走过,距离设为无穷大
dist[1] = 0;
queue<int> q; // 宽搜队列,宽搜深搜都是队列
q.push(1); // 1 号入队
while(q.size()) // 只要队列非空就一直往后搜
{
int t = q.front(); // 队头出队,找到该点能到的点
q.pop();
for(int i = h[t]; i != -1; i = ne[i])
{
int j = e[i];// 通过 伪指针 i 得到t能到的节点编号
if(!st[j]) // 如果该点没有遍历过
{
dist[j] = dist[i] + 1;
q.push(j); // 节点入队
st[j] = 1; // 标记
}
}
}
}
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); // 加入邻接表
}
bfs();
cout << (dist[n] == 0x3f3f3f3f ? -1 : dist[n]);
return 0;
}