相当于求从源点到其他所有点的最短路,只需要跑一次bfs更新每个点的邻接点即可,之前想要直接暴力结果超时了。(要用bfs求最短路必须满足所有边的权值都相同)
#include <iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
using namespace std;
const int M=2e5+7;
const int inf=1e9;
int a[M],d[M];
queue<int>q;
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
d[i]=inf;
}
q.push(1);
d[1]=0;
while(!q.empty()) //bfs
{
int x=q.front();
q.pop();
if(x<n&&d[x+1]>d[x]+1)
{
d[x+1]=d[x]+1;
q.push(x+1);
}
if(x>1&&d[x-1]>d[x]+1)
{
d[x-1]=d[x]+1;
q.push(x-1);
}
if(d[a[x]]>d[x]+1)
{
d[a[x]]=d[x]+1;
q.push(a[x]);
}
}
/*for(int i=1;i<=n;i++) //暴力
{
for(int j=1;j<i;j++)
{
if(a[j]==i)
b[i]=min(b[i],b[j]+1);
if(a[j]>i)
b[i]=min(b[i],b[j]+1+a[j]-i);
b[i]=min(b[i],b[j]+i-j);
}
}*/
for(int i=1;i<n;i++)
printf("%d ",d[i]);
printf("%d\n",d[n]);
return 0;
}