题意:给出一个三维迷宫,以及起点和终点。单位时间可以往上下左右前后走一格。询问从起点到终点的最短距离(可能不存在)。
思路:二维最短路的升级版,bfs搜索最短路径即可。实现时可以用pair结构记录相互关联的几个数据。
代码:
//#include<bits/stdc++.h>
#include<stdio.h>
#include<stdlib.h>
#include<queue>
#include<cstring>
#include<iostream>
using namespace std;
typedef long long ll;
typedef long double db;
typedef pair<int,int> pii;
typedef vector<int> vi;
#define de(x) cout << #x << "=" << x << endl
#define rep(i,a,b) for(int i=a;i<(b);++i)
#define all(x) (x).begin(),(x).end()
#define sz(x) (int)(x).size()
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define lb(x) (x&-(x))
const int N = 101010;
int ans;
int x,y,z,sx,sy,sz,ex,ey,ez,npx,npy,npz,nps,tmpx,tmpy,tmpz,tmps;
char a[33][33][33];
int vis[33][33][33];
int dir[6][3]={1,0,0, -1,0,0, 0,-1,0, 0,1,0, 0,0,1, 0,0,-1};
queue< pair< pair<int,int> , pair<int,int> > > q;
int chk(int k,int i,int j)
{
if(k>0&&i>0&&j>0&&k<=z&&i<=x&&j<=y&&vis[k][i][j]==0&&a[k][i][j]!='#') return 1;
else return 0;
}
void bfs(int z,int x,int y)
{
vis[z][x][y]=1;
pair< pair<int,int> , pair<int,int> > p=mp(mp(0,z),mp(x,y));
q.push(p);
while(!q.empty())
{
pair< pair<int,int> , pair<int,int> > np=q.front(); q.pop();
npx=np.second.first;npy=np.second.second;npz=np.first.second;nps=np.first.first;
//判断是不是答案
if(npx==ex&&npy==ey&&npz==ez)
{
ans=nps;
// de(nps);
return ;
}
//扩展队列
for(int i=0;i<6;i++)
{
tmpx=npx+dir[i][1];tmpy=npy+dir[i][2];tmpz=npz+dir[i][0];tmps=nps+1;
if(chk(tmpz,tmpx,tmpy))
{
vis[tmpz][tmpx][tmpy]=1;
// printf("insert %d %d %d step:%d\n",tmpx,tmpy,tmpz,tmps);
q.push(mp(mp(tmps,tmpz),mp(tmpx,tmpy)));
}
}
}
}
int main(){
while(scanf("%d%d%d",&z,&x,&y)&&x!=0&&y!=0&&z!=0)
{
ans=-1;
memset(a,0,sizeof(a));
memset(vis,0,sizeof(vis));
while(!q.empty()) q.pop();
rep(k,1,z+1) rep(i,1,x+1) rep(j,1,y+1)
{
scanf(" %c",&a[k][i][j]);
if(a[k][i][j]=='S') sx=i,sy=j,sz=k;
if(a[k][i][j]=='E') ex=i,ey=j,ez=k;
}
bfs(sz,sx,sy);
if(ans!=-1) printf("Escaped in %d minute(s).\n",ans);
else printf("Trapped!\n");
}
return 0;
}