链接:https://www.nowcoder.net/acm/contest/68/B
来源:牛客网
题目描述
这是一个关于二维迷宫的题目。我们要从迷宫的起点 'S' 走到终点 'E',每一步我们只能选择上下左右四个方向中的一个前进一格。 'W' 代表墙壁,是不能进入的位置,除了墙壁以外的地方都可以走。迷宫内的 'D' 代表一道上锁的门,只有在持有钥匙的时候才能进入。而 'K' 则代表了钥匙,只要进入这一格,就会自动地拿到钥匙。最后 '.' 则是代表空无一物的地方,欢迎自在的游荡。
本题的迷宫中,起点、终点、门跟钥匙这四个特殊物件,每一个恰好会出现一次。而且,此迷宫的四周 (最上面的一行、最下面的一行、最左边的一列以及最右边的一列) 都会是墙壁。
请问,从起点到终点,最少要走几步呢?
输入:
输入的第一行有两个正整数H, W,分别代表迷宫的长跟宽。
接下来的H行代表迷宫,每行有一个长度恰为W的字串,此字串只包含`'S'`, `'E'`, `'W'`, `'D '`, `'K'`, `'.'`这几种字元。
输出:
请在一行中输出一个整数代表答案,如果无法从起点走到终点,请输出-1。
样例输入:
4 12
WWWWWWWWWWWW
W E . W . S . . W . K W
W . . D . . . . . . . W
WWWWWWWWWWWW
6 6
WWWWWW
W E W S .W
W . W K .W
W . W D . W
W . W . . W
WWWWWW
样例输出:
20
-1
用BFS先算出S到E的最短距离,然后再算出S到K的最短距离和K到E的最短距离的和,然后两个距离比较,较小的就是最小步数。
代码:
代码比较low,将就着看吧。。。。。
#include<stdio.h>
#include<algorithm>
#include<stdlib.h>
#include<malloc.h>
#include<math.h>
#include<string>
#include<queue>
#include<stack>
#include<cstring>
#include<vector>
#define inf 0x3f3f3f3f
#define N 1000000+10
#define LL long long
#define MAX 1000000007
#define pi 3.1415926
using namespace std;
int flag,h,w,min1=inf,min2=inf,min3=inf;
int num,vis[1000][1000],dp[600][600];
struct node{
char c;
int x,y,s;
}s[600][600];
void bfs1(int x,int y)
{
queue<struct node> q;
q.push(s[x][y]);
vis[x][y]=1;
for(int i=0;i<=500;i++)
for(int j=0;j<=500;j++)
dp[i][j]=inf;
dp[x][y]=0;
while(!q.empty())
{
struct node u=q.front();
q.pop();
if(u.c=='E')
{
min1=dp[u.x][u.y];
}
x=u.x;
y=u.y;
if((s[x+1][y].c!='D'&&s[x+1][y].c!='W')&&vis[x+1][y]==0)
q.push(s[x+1][y]),dp[x+1][y]=dp[x][y]+1;
if((s[x-1][y].c!='D'&&s[x-1][y].c!='W')&&vis[x-1][y]==0)
q.push(s[x-1][y]),dp[x-1][y]=dp[x][y]+1;
if((s[x][y+1].c!='D'&&s[x][y+1].c!='W')&&vis[x][y+1]==0)
q.push(s[x][y+1]),dp[x][y+1]=dp[x][y]+1;
if((s[x][y-1].c!='D'&&s[x][y-1].c!='W')&&vis[x][y-1]==0)
q.push(s[x][y-1]),dp[x][y-1]=dp[x][y]+1;
vis[x][y]=1;
}
}
void bfs2(int x,int y)
{
queue<struct node>q;
q.push(s[x][y]);
vis[x][y]=1;
for(int i=0;i<=500;i++)
for(int j=0;j<=500;j++)
dp[i][j]=inf;
dp[x][y]=0;
while(!q.empty())
{
struct node u=q.front();
q.pop();
if(u.c=='K')
{
min2=dp[u.x][u.y];
}
x=u.x;
y=u.y;
if((s[x+1][y].c!='D'&&s[x+1][y].c!='W')&&vis[x+1][y]==0)
q.push(s[x+1][y]),dp[x+1][y]=dp[x][y]+1;
if((s[x-1][y].c!='D'&&s[x-1][y].c!='W')&&vis[x-1][y]==0)
q.push(s[x-1][y]),dp[x-1][y]=dp[x][y]+1;
if((s[x][y+1].c!='D'&&s[x][y+1].c!='W')&&vis[x][y+1]==0)
q.push(s[x][y+1]),dp[x][y+1]=dp[x][y]+1;
if((s[x][y-1].c!='D'&&s[x][y-1].c!='W')&&vis[x][y-1]==0)
q.push(s[x][y-1]),dp[x][y-1]=dp[x][y]+1;
vis[x][y]=1;
}
}
void bfs3(int x,int y)
{
queue<struct node>q;
q.push(s[x][y]);
vis[x][y]=1;
for(int i=0;i<=500;i++)
for(int j=0;j<=500;j++)
dp[i][j]=inf;
dp[x][y]=0;
while(!q.empty())
{
struct node u=q.front();
q.pop();
if(u.c=='E')
{
min3=dp[u.x][u.y];
}
x=u.x;
y=u.y;
if(s[x+1][y].c!='W'&&vis[x+1][y]==0)
q.push(s[x+1][y]),dp[x+1][y]=dp[x][y]+1;
if((s[x-1][y].c!='W')&&vis[x-1][y]==0)
q.push(s[x-1][y]),dp[x-1][y]=dp[x][y]+1;
if((s[x][y+1].c!='W')&&vis[x][y+1]==0)
q.push(s[x][y+1]),dp[x][1+y]=dp[x][y]+1;
if((s[x][y-1].c!='W')&&vis[x][y-1]==0)
q.push(s[x][y-1]),dp[x][y-1]=dp[x][y]+1;
vis[x][y]=1;
}
}
int main()
{
while(~scanf("%d%d",&h,&w))
{
int x1,y1,x2,y2;
flag=0;
min1=min2=min3=inf;
getchar();
memset(vis,0,sizeof(vis));
memset(s,0,sizeof(s));
for(int i=0;i<h;i++)
{
for(int j=0;j<w;j++)
{
scanf("%c",&s[i][j].c);
s[i][j].x=i;
s[i][j].y=j;
if(s[i][j].c=='S')
x1=i,y1=j;
if(s[i][j].c=='K')
x2=i,y2=j;
}
getchar();
}
vis[x1][y1]=1;
bfs1(x1,y1);
memset(vis,0,sizeof(vis));
vis[x1][y1]=1;
bfs2(x1,y1);
if(min2!=inf)
flag=1;
if(flag==1)
{
memset(vis,0,sizeof(vis));
vis[x2][y2]=1;
bfs3(x2,y2);
}
/*printf("%d %d %d\n",min1,min2,min3);*/
int ans=min(min1,min2+min3);
if(ans>=inf)
printf("-1\n");
else
printf("%d\n",ans);
}
return 0;
}