树与图的广度优先遍历

题目

 

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

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

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

输入格式

第一行包含两个整数 nn 和 mm。

接下来 mm 行，每行包含两个整数 aa 和 bb，表示存在一条从 aa 走到 bb 的长度为 11 的边。

输出格式

输出一个整数，表示 11 号点到 nn 号点的最短距离。

1、存树

第一步先将数用链表存起来；

#include <cstring>
#include <iostream>

using namespace std;

const int N=1e5+10;

int h[N], e[N], idx, ne[N];
int d[N]; //存储每个节点离起点的距离  d[1]=0
int n, m; //n个节点m条边
int q[N]; //存储层次遍历序列 0号节点是编号为1的节点
void add(int a, int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

解释：

 h[i]用来存节点，它的子节点用链表形式存起来，e[ ]存子节点；

2、队列的运用

宽度搜索需要借助队列，在遍历层的时候将节点入队，q[0] = 1，首先要将1节点存入第一个位置；

每次取出队头，在队尾加入节点；

3、宽度搜索

第二步开始遍历，进行宽度搜索。

从一层开始，逐层搜索；

 int bfs()
{
    q[0] = 1;
    int tt = 0,hh = 0;
    memset(d,-1,sizeof d);
    d[1] = 0;
    while(tt <= hh)
{
    int t = q[hh ++];
    for(int i = h[t];i != -1;i = ne[i])
    {
        int j = e[i];
        if(d[j]!=-1)
        {
            d[i] = d[t] + 1;
            q[++ tt] = e[i];
        }
    }
}
    return d[n];
}

4、代码整体

#include <cstring>
#include <iostream>

using namespace std;

const int N=1e5+10;

int h[N], e[N], idx, ne[N];
int d[N]; //存储每个节点离起点的距离  d[1]=0
int n, m; //n个节点m条边
int q[N]; //存储层次遍历序列 0号节点是编号为1的节点
void add(int a, int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
} 

int bfs()
{
    q[0] = 1;
    int tt = 0,hh = 0;
    memset(d,-1,sizeof d);
    d[1] = 0;
    while(tt <= hh)
{
    int t = q[hh ++];
    for(int i = h[t];i != -1;i = ne[i])
    {
        int j = e[i];
        if(d[j]!=-1)
        {
            d[i] = d[t] + 1;
            q[++ tt] = e[i];
        }
    }
}
    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;
}