第二次训练赛 B. Mike and Shortcuts-bfs

题目传送门：https://codeforces.com/problemset/problem/689/B

题目大意：有n个城市在一排，两两间距为1，每个城市可以用1的距离到一个给定的城市，问城市1到其他所有城市的最短距离。

思路：bfs。

比赛时脑抽写了个dijkstra，还wa了。。。
bfs的时候一定要搜到一个标记一个。。。

代码：

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll M=2e5+10;
bool vis[M];
ll n,a[M],d[M];
void bfs(){
queue<ll> q;
q.push(1);
vis[1]=1;
while(!q.empty()){
ll p=q.front();
q.pop();
if(p<n && !vis[p+1]) d[p+1]=d[p]+1,vis[p+1]=1,q.push(p+1);
if(p>1 && !vis[p-1]) d[p-1]=d[p]+1,vis[p-1]=1,q.push(p-1);
if(a[p] && !vis[a[p]]) d[a[p]]=d[p]+1,vis[a[p]]=1,q.push(a[p]);
}
}
int main(){
scanf("%lld",&n);
for(ll i=1;i<=n;i++) scanf("%lld",&a[i]);
bfs();
for(ll i=1;i<=n;i++) printf("%lld ",d[i]);
return 0;
}