每日打卡(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了。
//为了蓝桥杯,目前一阶段的练习以暴力搜索为主