图中点的层次——树与图的广度优先遍历

问题描述

代码实现

#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10;

int n, m;
int h[N], ne[N * 2], e[N * 2], idx;
int d[N];	// 从节点1到当前节点的距离
int q[N * 2];	// 数组模拟队列

void add(int a, int b)	// 邻接表存储图
{
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void bfs()
{
	memset(d, -1, sizeof d);	// 将从节点1到每个节点距离初始化为-1
	int hh = 0, tt = 0;	// 队头、队尾指针
	q[hh] = 1;	// 将节点1从队尾插入
	d[1] = 0;	// 节点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)	// 如果没有遍历过，加入队列中，并跟新节点1到节点j的距离
			{
				d[j] = d[t] + 1;
				q[++tt] = j;
			}
		}
	}
	
	cout << d[n] << endl;
}

int main()
{
	memset(h, -1, sizeof h);
	cin >> n >> m;
	for(int i = 0; i < m; i++)
	{
		int a, b;
		cin >> a >> b;
		add(a, b);
	}
	bfs();
}