计蒜客习题: 最短路简化版

#include <bits/stdc++.h>
using namespace std;
const int maxn=1010;
vector<int> p[maxn]; //建图
int d[maxn]; //距离
queue <int >q;

void init(){
memset(d,-1,sizeof(d));   //d=-1时代表尚未经过该边
}

void bfs(int cur){
q.push(cur);
d[cur]=0;
while(!q.empty()){
    int tmp=q.front();
    q.pop();
    int sz=p[tmp].size();
    for(int i=0;i<sz;i++){
        if(d[p[tmp][i]]==-1){
            q.push(p[tmp][i]);
            d[p[tmp][i]]=d[tmp]+1;
        }
    }
}
return ;
}

int main(){
init();
int m,n,c;
cin>>n>>m>>c;
for(int i=0;i<m;i++){
    int a,b;
    cin>>a>>b;
    p[a].push_back(b);
    p[b].push_back(a);
}
bfs(c);
for(int i=1;i<=n;i++){
    cout<<d[i]<<endl;
}
return 0;
}