2019 GDUT Winter Personal Training Contest III (Div. 2) B. Mike and Shortcuts
题目见 :
http://codeforces.com/group/NVaJtLaLjS/contest/236618/problem/B
题目大意:
给出n个格子,每次可以从当前位置第i格往前或后走一格,或者走到ai格(ai>i),求从1走到所有格子的最短步数。(n<=200 000)
思路:
以前好像在洛谷上刷过类似的题,很明显是个bfs。
一次优化:
注意!bfs里面迭代的顺序非常重要,我第一次先往后走,结果超时。因为题目有条件ai>i,那么如果能够跳跃走,优先走到ai格,这是最节约步数的结合贪心。
实现:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
int f[200002],a[200002];
queue<int>q;//bfs队列
int main()
{
int i,j,n;
scanf("%d",&n);
for (i=1;i<=n;i++)
{
f[i]=n+1;
scanf("%d",&a[i]);
}
q.push(1);
f[1]=0;
while (q.empty()==false)
{
//printf(" %d %d %d\n",q.front(),f[q.front()],q.back());
// scanf("%d",&j);
if (a[q.front()]<=n&&f[a[q.front()]]==n+1)//能跳则跳
{
f[a[q.front()]]=f[q.front()]+1;
q.push(a[q.front()]);
}
if (q.front()+1<=n&&f[q.front()+1]==n+1)
{
f[q.front()+1]=f[q.front()]+1;
q.push(q.front()+1);
}
if (q.front()-1>=1&&f[q.front()-1]==n+1)
{
f[q.front()-1]=f[q.front()]+1;
q.push(q.front()-1);
}
q.pop();
}
for (i=1;i<=n-1;i++)
printf("%d ",f[i]);
printf("%d\n",f[n]);
return 0;
}