这道题运用bfs的方法,首先每一层可以摁up或者down,所以每一个节点可以扩展出两个节点。先把起始层节点入队,为了防止在之后重复拓展此节点,将在标记为true,之后在提取出头结点,并将其删除。同时将其的两个子节点入队。重复以上操作。如果最后出现节点==B(目标层数),则寻找成功,返回time【next】,函数调用结束,否则如果执行到最后,一直没有到达目标层数,则返回-1。这里要特别注意一点,就是,一开始我把K[maxn]下标从0开始定义了,出现了致命错误。题意第一层为k1,第二层为k2, 函数返回值,被用来被调函数返回主调函数继续执行。
#include<stdio.h>
#include<iostream>
#include<queue>
#include<string.h>
#define maxn 200+5
bool visit[maxn],flag;
int time[maxn],K[maxn];
using namespace std;
queue <int> q;
int bfs(int A,int B)
{
int head,next;
q.push(A);
visit[A]=true;
time[A]=0;
while(!q.empty())
{
head=q.front();
q.pop();
for(int i=0;i<2;i++)
{
if(i==0)
next=head+K[head];
if(i==1)
next=head-K[head];
if(next<0||next>200)
continue;
if(!visit[next])
{
q.push(next);
time[next]=time[head]+1;
visit[next]=true;
}
if(next==B)
return time[next];
}
}
return -1;
}
int main()
{
int N,A,B,t;
while(scanf("%d",&N)==1&&N)
{
bool flag=true;
int t;
scanf("%d%d",&A,&B);
memset(visit,false,sizeof(visit));
memset(time,0,sizeof(time));
for(int i=1;i<=N;i++)
scanf("%d",&K[i]);
if(A==B)
{
printf("0\n");
continue;
}
while(!q.empty()) q.pop();
t=bfs(A,B);
printf("%d\n",t);
}
return 0;
}