【BFS】奇怪的电梯

题目描述

大楼的每一层楼都可以停电梯，而且第i层楼（1<=i<=N）上有一个数字Ki（0<=Ki<=N）。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如：3 3 1 2 5代表了Ki（K1=3,K2=3,……），从一楼开始。在一楼，按“上”可以到4楼，按“下”是不起作用的，因为没有-2楼。那么，从A楼到B楼至少要按几次按钮呢？

输入

输入文件共有二行，第一行为三个用空格隔开的正整数，表示N,A,B(1≤N≤200, 1≤A,B≤N)，第二行为N个用空格隔开的正整数，表示Ki。

输出

输出文件仅一行，即最少按键次数,若无法到达，则输出-1。

样例输入

5 1 5
3 3 1 2 5

样例输出

3

思路：

广搜思路，一层一层的搜。

#include<iostream>
#include<cstdio>
using namespace std;
int head,tail,k,n,m,a[500],d[500][3];
bool f[500]={0};
int main()
{
	scanf("%d%d%d",&k,&n,&m);
	if (n==m)//判断是否在目标层
	{
		printf("0");
		return 0;
	}
	for(int i=1;i<=k;i++)
		scanf("%d",&a[i]);
	d[1][1]=n;
	d[1][2]=a[n];
	d[1][3]=0;
	head=tail=1;//定义初始值
	while(head<=tail)
	{
		if (d[head][1]+d[head][2]<=k)//判断是否出界
		{
			tail++;
			d[tail][1]=d[head][1]+d[head][2];
			d[tail][2]=a[d[tail][1]];
			d[tail][3]=d[head][3]+1;//此路径加一
		    if (f[d[tail][1]])tail--;//判断是否走过
		    else f[d[tail][1]]=1;
		    if (d[tail][1]==m)//判断是否到达目标层
		    {
			    printf("%d",d[tail][3]);
			    return 0;
		    };
		};
		
		
		if (d[head][1]-d[head][2]>0)//判断是否出界
		{
			tail++;
			d[tail][1]=d[head][1]-d[head][2];
			d[tail][2]=a[d[tail][1]];
			d[tail][3]=d[head][3]+1;//此路径加一
			if (f[d[tail][1]])tail--;//判断是否走过
		    else f[d[tail][1]]=1;
		    if (d[tail][1]==m)//判断是否到达目标层
		    {
		    	printf("%d",d[tail][3]);
			    return 0;
		    };
		};
		head++;
		
	}
	printf("-1\n");//如果没有结果就输出-1
	return 0;
}