ACwing847. 图中点的层次

题目

分析

所有边的长度都是1，权值相同，用BFS求最短路

代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int N = 100010;
int h[N],e[N],ne[N],idx;//有向图，e,en不用N乘2
int n,m;//图的结点和边
int d[N];//记录从1号到各个点的距离

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

int bfs(){
    queue<int>q;
    q.push(1);
    d[1] = 0;
    
    while(!q.empty()){
        int t = q.front();
        q.pop();
        for(int i = h[t];i!=-1;i = ne[i]){
            int j = e[i];
            if(d[j] == -1){ //检查是否已经访问过
                d[j] = d[t]+1;
                q.push(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);
    }
    
    memset(d,-1,sizeof(d));
    printf("%d",bfs());
    return 0;
}

对于图的存储，一定别忘记对 h 的初始化第37行