【AcWing——宽度优先搜索】AcWing 847. 图中点的层次遍历

AcWing 847. 图中点的层次遍历

使用宽度优先搜索，队列使用数组进行模拟。

#include <iostream>
#include <cstring>

using namespace std;

const int N = 100010;
int n, m;
int h[N], e[N], ne[N], idx = 0;//用于图的邻接表存储
int d[N], q[N];//d用来表示距离，q用来数组模拟队列

void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

int bfs() {
    int hh = 0, tt = 0;//hh为队头， tt为队尾
    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() {
    scanf("%d%d", &n, &m);
    
    memset(h, -1, sizeof h);
    
    for (int i = 0; i < m; ++i) {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b);
    }
    
    printf("%d\n", bfs());
    
    return 0;
}