迷宫(bfs)

链接: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;
}


猜你喜欢

转载自blog.csdn.net/qwerqwee12/article/details/79306511