计蒜客 最短路简化版

转载 | 原文链接

问题描述

经历一周忙碌的工作后，蒜头君想趁着周末好好游玩一番。蒜头君想去好多好多地方，他想去南锣鼓巷吃各种好吃的，想去颐和园滑冰，还想去怀柔滑雪场滑雪……可是时间有限，蒜头君并不能玩遍所有的地方，最后他决定去几个离他较近的。
我们知道蒜头君一共想去 N 个地方玩耍，编号从 1 到 N，并且知道了蒜头君所在地方的编号 C，以及 M 条路径。现在蒜头君想让你帮他算一算，他到每个地方分别需要经过多少个地方？
输入格式
第一行输入三个正整数 N,M,C。代表蒜头君想去 N 个地方，有 M 条路径，蒜头君在编号为 C 的地方。1≤N,C≤1000, 1≤C≤N, 1≤M≤10000。
保证没有重复边，且图中所有点互相连通。
输出格式
输出 N 行，按编号从小到大，输出结果。第 i 行表示蒜头君到编号为 i 的地方，需要经过多少个地方。

样例输入

5 5 2 
1 2 
2 3 
2 4 
3 4 
3 5

样例输出

1 
0 
1 
1 
2

AC代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
vector<int>biao[1010];
int d[1010];
queue<int>q;

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

int main()
{
    memset(d,-1,sizeof(d));
    int m,n,c;
    cin>>n>>m>>c;
    while(m--)
    {
        int a,b;
        cin>>a>>b;
        biao[a].push_back(b);
        biao[b].push_back(a);
    }
    bfs(c);
    for(int i=1;i<=n;i++)
    {
        cout<<d[i]<<"\n";
    }
    return 0;
}