P1135 奇怪的电梯(BFS)

题目描述

呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第ii层楼(1 \le i \le N)(1≤i≤N)上有一个数字K_i(0 \le K_i \le N)Ki​(0≤Ki​≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3, 3 ,1 ,2 ,53,3,1,2,5代表了K_i(K_1=3,K_2=3,…)Ki​(K1​=3,K2​=3,…),从11楼开始。在11楼,按“上”可以到44楼,按“下”是不起作用的,因为没有-2−2楼。那么,从AA楼到BB楼至少要按几次按钮呢?

输入格式

共二行。

第一行为33个用空格隔开的正整数,表示N,A,B(1≤N≤200, 1≤A,B≤N)N,A,B(1≤N≤200,1≤A,B≤N)。

第二行为NN个用空格隔开的非负整数,表示K_iKi​。

输出格式

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

输入输出样例

输入 #1复制

5 1 5
3 3 1 2 5

输出 #1复制

3

我用的是path数组来存放已经用的步数 下一个需要遍历的结点的步数就等于前一结点的步数+1

#include<iostream>
#include<queue>
using namespace std;
int k[205],N,A,B,now,flag=0,path[205],next1,next2;
bool vis[205];
void bfs(int start)
{
	queue<int>q;
	vis[start]=true;	//把初始点标记一下 
	path[start]=0;
	q.push(start);
	while(!q.empty())
	{
		 now=q.front();
		 q.pop();
		 if(now==B){
		 	cout<<path[now];		//如果到达了终点就输出path[终点],并且标记以下 如果整个遍历完了flag==0,那么就是不能到达 输出-1 
		 	flag=1;
		 	return ;
		 }
		 next1=now+k[now];
		 next2=now-k[now];
		 if(!vis[next1]&&next1>=1&&next1<=N)
		 {
		 	vis[next1]=true;
		 	q.push(next1);
		 	path[next1]=path[now]+1;
		 }
		 if(!vis[next2]&&next2>=1&&next2<=N)
		 {
		 	vis[next2]=true;
		 	q.push(next2);
		 	path[next2]=path[now]+1;
		 }
	}
}
int main()
{
	cin>>N>>A>>B;
	for(int i=1;i<=N;i++)
	{
		cin>>k[i];
	}
	bfs(A);
	if(flag==0)		//不能到达终点 输出-1  第17行有判断的解释 
	cout<<"-1";
}

猜你喜欢

转载自blog.csdn.net/qq_52245648/article/details/119768945