算法学习专栏——广度优先遍历
一、图中点的层次(中等-)
给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环。
所有边的长度都是 1,点的编号为 1∼n。
请你求出 1 号点到 n 号点的最短距离,如果从 1 号点无法走到 n 号点,输出 −1。
输入格式
第一行包含两个整数 n 和 m。
接下来 m 行,每行包含两个整数 a 和 b,表示存在一条从 a 走到 b 的长度为 1 的边。
输出格式
输出一个整数,表示 1 号点到 n 号点的最短距离。
数据范围
1≤n,m≤10^5^
输入样例:
1 | |
输出样例:
1 | |
思路
代码
答案(请自己先思考一下再参考)
#include < iostream>
#include < cstring>
using namespace std;
const int N = 1e5 + 10;
int n, m;
int e[N],ne[N],h[N],idx;
int q[N],d[N];
void add(int x,int y)
{
e[idx] = y;ne[idx] = h[x],h[x] = idx++;
}
int bfs(int x)
{
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 j = e[i];
if (d[j] == -1)
{
d[j] = d[t] + 1;
q[++tt] = j;
}
}
}
return d[n];
}
int main()
{
ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
cin >> n >> m;
memset(h, -1, sizeof h);
for (int i = 0; i < m; i++)
{
int x, y;
cin >> x >> y;
add(x, y);
}
cout << bfs(1);
return 0;
}