洛谷 P1135 奇怪的电梯

X BoHang

于 2024-03-18 20:55:12 发布

阅读量392

点赞数 9

CC 4.0 BY-SA版权

文章标签：算法

本文链接：https://blog.youkuaiyun.com/xiebohang/article/details/136821436

这篇文章介绍了一个编程问题，通过C++代码实现广度优先搜索(BFS)，解决在多层楼中，从给定的起点a到终点b，最少乘坐电梯次数的问题。代码定义了结构体和队列，利用k数组存储每层电梯数量，vis数组标记已访问的楼层。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码如下:

#include<bits/stdc++.h>
using namespace std;
const int N=202;
int k[N],vis[N];		//k表示每层电梯上的数字；vis：标记数组；
int n,a,b,res=-1;		//res表示结果；
struct node{
	int floor,step;		//floor：当前位于第几层；step：从a层到达该层需要按的电梯次数； 
};
int f[]={-1,1};		//-1：向下；1:向上；
queue<node> q;
void bfs();
int main(){
	scanf("%d %d %d",&n,&a,&b);
	for(int i=1;i<=n;i++){
		scanf("%d",&k[i]);
	}
	bfs();
	printf("%d\n",res);
	return 0;
} 
void bfs(){
	//先判断起点和终点是否重合。 
	if(a==b){
		res=0;
		return;
	}
	vis[a]=1;
	node nod={a,0};
	q.push(nod);
	while(!q.empty()){
		nod=q.front();
		q.pop();
		int floor=nod.floor;
		int step=nod.step;
		for(int i=0;i<=1;i++){
			int nf=floor+(k[floor]*f[i]);
			if(nf<1 || nf>n){
				continue;
			}
			if(vis[nf]){
				continue;
			}
			if(nf==b){
				res=step+1;
				return;
			}
			vis[nf]=1;
			node nnod={nf,step+1};
			q.push(nnod);
		}
	}
}