链接: 原题链接
输入输出样例
输入 #1
5 1 5
3 3 1 2 5
输出 #1
3
思路讲解
本题有很多做法,BFS、DFS、floyd、spfa等等,下面以BFS和DFS为例。
思路很简单,从起点开始,只要没越界就向上下搜,全部搜完得到答案。
注意:DFS需要剪枝,不然会TLE或MLE。
代码实现
BFS:
#include<bits/stdc++.h>
using namespace std;
const int MAXN=205;
int n,st,ed;
int k[MAXN];
int v[MAXN];
int minn=99999999;
int flag;
int d[2]={1,-1};
void BFS(int floor,int ct);
struct node{
int f,ct;
} start, temp;;
queue<node> r;
int main()
{
scanf("%d%d%d",&n,&st,&ed);
for(int i=1;i<=n;i++)
scanf("%d",&k[i]);
BFS(st,0);
if(flag) printf("%d",minn);
else printf("-1");
}
void BFS(int floor,int ct)
{
v[floor]=1;
start.f=floor;
start.ct=ct;
r.push(start);
while(!r.empty())
{
int rf=r.front().f;
// if(r.front().ct>minn)continue;
if(rf==ed)
{
flag=1;
if(minn>r.front().ct)
minn=r.front().ct;
break;
}
for(int i=0;i<=1;i++)
{
temp.f=rf+k[rf]*d[i];
temp.ct=r.front().ct+1;
if(temp.f>n||temp.f<1) continue;
if(v[temp.f]==0)
{
r.push(temp);
v[temp.f]=1;
}
}
r.pop();
}
}
DFS:
#include<bits/stdc++.h>
using namespace std;
const int MAXN=205;
int n,st,ed;
int k[MAXN],v[MAXN];
int ans=99999,flag;
int d[2]={1,-1};//上下方向
void DFS(int floor,int ct); //在floor层,已经按了ct次
int main()
{
scanf("%d%d%d",&n,&st,&ed);
for(int i=1;i<=n;i++)
scanf("%d",&k[i]);
flag=0;
v[st]=1;
DFS(st,0);
if(flag) printf("%d",ans);
else printf("-1");
}
void DFS(int floor,int ct) //在floor层,已经按了ct次
{
if(floor==ed)
{
flag=1;
if(ct<ans)
ans=ct;
return;
}
if(ct>ans)return;//剪枝
int tf;
v[floor]=1;
for(int i=0;i<=1;i++)
{
tf=floor+k[floor]*d[i];
if(tf>n||tf<1)continue;
if(v[tf]==0)
{
DFS(tf,ct+1);
}
}
v[floor]=0;
}