题意:
有n个路口,第i个路口到第j个路口需要abs(i-j)的体力,并且每个路口有一条单向捷径a[i]直接到路口a[i]消耗1点体力,求从路口1到其他所有路口所需最小体力
题解:
第i个路口到第j个路口消耗体力abs(i-j),因为对于每个路口可以花费1点体力向i+1或者i-1个路口移动,所以从i个路口到第j个路口的一次移动可以视为abs(i-j)次临近移动,然后就可以用bfs求解了
#include<iostream>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include<stdlib.h>
#include <string.h>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<time.h>
using namespace std;
#define MAX_N 200005
#define inf 0x7fffffff
#define LL long long
#define ull unsigned long long
#define mod 1000000007
LL INF=9e18;
int n;
int a[MAX_N];
int ans[MAX_N];
bool vis[MAX_N];
struct node
{
int x,t;
};
void bfs()
{
node now, next;
queue<node> q;
fill(ans+1, ans+1+n, inf);
now.x = 1; now.t = 0;
vis[1] = true;
ans[1] = 0;
q.push(now);
while(!q.empty()) {
now = q.front();
q.pop();
for(int i=-1;i<=1;i++) {
if(!i) {
next.x = a[now.x];
next.t = now.t + 1;
if(!vis[next.x]) {
vis[next.x] = true;
ans[next.x] = next.t;
q.push(next);
}
}
else {
next.x = now.x + i;
next.t = now.t + 1;
if(next.x>=1 && next.x<=n && !vis[next.x]) {
vis[next.x] = true;
ans[next.x] = next.t;
q.push(next);
}
}
}
}
}
int main()
{
cin >> n;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
bfs();
for(int i=1;i<=n;i++)
printf("%d%c",ans[i],i==n?'\n':' ');
}