将可以直接到达目标城市的答案(cnt)设置成1,由该点往外扩,每次更新队列。
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+10;
const int M=2e5+10;
int n,m,k;
int a;
int cnt[N];
bool pd[N];
struct re{
int to;
int next;
}e[M*2];
int h[N];
int idx;
struct ee{
int num;
int from;
};
queue<ee> q;
void add(int x,int y){
e[++idx].to=y;
e[idx].next=h[x];
h[x]=idx;
}
signed main(){
cin>>n>>m>>k;
memset(h,-1,sizeof h);
memset(cnt,-1,sizeof cnt);
for(int i=1;i<=k;i++) cin>>a,pd[a]=1;
for(int i=1;i<=m;i++){
int x,y;
cin>>x>>y;
if(x==y) continue;
if(pd[x]) cnt[y]=1,q.push((ee){y,x});
if(pd[y]) cnt[x]=1,q.push((ee){x,y});
add(x,y);
add(y,x);
}
while(!q.empty()){
int x=q.front().num;
int from=q.front().from;
q.pop();
for(int i=h[x];i!=-1;i=e[i].next){
int to=e[i].to;
if(to!=from&&(cnt[to]==-1||cnt[to]+1<cnt[x])){
cnt[to]=cnt[x]+1;
q.push((ee){to,x});
}
}
}
for(int i=1;i<=n;i++) cout<<cnt[i]<<" ";
return 0;
}