POJ2251 3D迷宫（BFS）

每日打卡（2/2）

点击打开链接

题目大意：

       一个三维的迷宫，给定起点和终点，求最短路径。

        ps：长宽高均不超过30

思路1：

    首先想到DFS，只是第一次写三维的DFS，感觉和二维没什么不一样。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int INF = 1e8+7;
const int maxn = 35;
bool vis[maxn][maxn][maxn];
int n,k,m,sx,sy,sz,fx,fy,fz;
int a[maxn][maxn][maxn];
long long ans,tot;
int dx[6]={1,0,0,0,0,-1};
int dy[6]={0,1,0,0,-1,0};
int dz[6]={0,0,1,-1,0,0};
void dfs(int x,int y,int z)
{
	if(x==fx&&y==fy&&z==fz) {
		ans = min(ans,tot);
		return;
	}
	if(ans!=INF&&tot>ans) return;
	for(int i=0;i<6;i++)
	{
		int newx = x+dx[i];
		int newy = y+dy[i];
		int newz = z+dz[i];
		if(newx>=0&&newx<n&&newy>=0&&newy<k&&newz>=0&&newz<m&&!vis[newx][newy][newz]&&a[newx][newy][newz])
		{
			vis[newx][newy][newz] = 1;
			tot++;
			dfs(newx,newy,newz);
			tot--;
			vis[newx][newy][newz] = 0;
		}
	}
}
int main()
{
	while(~scanf("%d%d%d",&n,&k,&m)&&n&&k&&m)
	{
		
		memset(a,0,sizeof(a));
		memset(vis,0,sizeof(vis));
		ans = INF;
		tot = 0;
		for(int i=0;i<n;i++)
		 for(int j=0;j<k;j++)
		  for(int l=0;l<m;l++)
		  {
		  	 char s;	
		  	 cin>>s;
		  	 if(s=='#') a[i][j][l] = 0;
		  	 else if(s=='.') a[i][j][l] = 1;
		  	 else if(s=='S') {
		  	 	sx = i,sy = j,sz = l;
		  	 	a[i][j][l] = 0;
		  	 }
		  	 else if(s=='E') {
		  	 	fx = i,fy = j,fz = l;
		  	 	a[i][j][l] = 1;
		  	 }
		  }
		
		dfs(sx,sy,sz);
		if(ans==INF) {
			cout<<"Trapped!"<<endl;
		}
		else
			cout<<"Escaped in "<<ans<<" minute(s)."<<endl;
	}
	return 0;
}

第一次写的时候吸取上午的经验，先剪了个枝。如果当前的tot值已经超过了之前的最优解，那么就pass。上次也遇到过这样一道题，这么剪枝依然TLE了。。果不其然，T了。这时候还有一种剪枝思路，就是每一步都记录到达当前格子的最优解，不满足直接pass，很可惜，我不会写。

思路二：

    DFS失败后，自然想到BFS。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>
#include<algorithm>
#include<queue> 
using namespace std;
const int INF = 1e8+7;
const int maxn = 35;
struct node{
	int x,y,z,tot;
};
bool vis[maxn][maxn][maxn];
int n,k,m,sx,sy,sz,fx,fy,fz;
int flag[maxn][maxn][maxn];
long long ans;
int dx[6]={1,0,0,0,0,-1};
int dy[6]={0,1,0,0,-1,0};
int dz[6]={0,0,1,-1,0,0};

int bfs()
{
	node a;
	queue<node> q;
	a.x = sx;
	a.y = sy;
	a.z = sz;
	a.tot = 0;
	q.push(a);
	while(!q.empty())
	{
		a = q.front();
		q.pop();
		if(a.x==fx&&a.y==fy&&a.z==fz)
        	return a.tot;
		for(int i=0;i<6;i++)
		{
			node next;
			next.x = a.x+dx[i];
			next.y = a.y+dy[i];
			next.z = a.z+dz[i];
			if(next.x>=0&&next.x<n&&next.y>=0&&next.y<k&&next.z>=0&&next.z<m&&!vis[next.x][next.y][next.z]&&flag[next.x][next.y][next.z])
			{
				next.tot = a.tot+1;
				q.push(next);
				vis[next.x][next.y][next.z]=1;
			}
		}
	}
	return 0;
}
int main()
{
	while(~scanf("%d%d%d",&n,&k,&m)&&n&&k&&m)
	{
		
		memset(vis,0,sizeof(vis));
		for(int i=0;i<n;i++)
		 for(int j=0;j<k;j++)
		  for(int l=0;l<m;l++)
		  {
		  	 char s;	
		  	 cin>>s;
		  	 if(s=='#') flag[i][j][l] = 0;
		  	 else if(s=='.') flag[i][j][l] = 1;
		  	 else if(s=='S') {
		  	 	sx = i,sy = j,sz = l;
		  	 	flag[i][j][l] = 0;
		  	 }
		  	 else if(s=='E') {
		  	 	fx = i,fy = j,fz = l;
		  	 	flag[i][j][l] = 1;
		  	 }
		  }
		
		ans = bfs();
		if(ans==0) {
			cout<<"Trapped!"<<endl;
		}
		else
			cout<<"Escaped in "<<ans<<" minute(s)."<<endl;
	}
	return 0;
}

由于宽搜的最优性，第一个得到的解一定是最优解，所以直接输出答案即可。

在宽搜过程中，往常我一般不会使用vis数组来标记，但是这次TLE了。吸取上次程序设计大赛C题买戒指的经验，加了个vis数组，果然A了。

//为了蓝桥杯，目前一阶段的练习以暴力搜索为主