牛客1007迷宫

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

这是一个关于二维迷宫的题目。我们要从迷宫的起点 'S' 走到终点 'E'，每一步我们只能选择上下左右四个方向中的一个前进一格。 'W' 代表墙壁，是不能进入的位置，除了墙壁以外的地方都可以走。迷宫内的 'D' 代表一道上锁的门，只有在持有钥匙的时候才能进入。而 'K' 则代表了钥匙，只要进入这一格，就会自动地拿到钥匙。最后 '.' 则是代表空无一物的地方，欢迎自在的游荡。

本题的迷宫中，起点、终点、门跟钥匙这四个特殊物件，每一个恰好会出现一次。而且，此迷宫的四周 (最上面的一行、最下面的一行、最左边的一列以及最右边的一列) 都会是墙壁。

请问，从起点到终点，最少要走几步呢？

输入描述:

输入的第一行有两个正整数H, W，分别代表迷宫的长跟宽。
接下来的H行代表迷宫，每行有一个长度恰为W的字串，此字串只包含`'S'`, `'E'`, `'W'`, `'D '`, `'K'`, `'.'`这几种字元。

输出描述:

请在一行中输出一个整数代表答案，如果无法从起点走到终点，请输出-1。

示例1

输入

复制4 12 WWWWWWWWWWWW WE.W.S..W.KW W..D..W....W WWWWWWWWWWWW

4 12
WWWWWWWWWWWW
WE.W.S..W.KW
W..D..W....W
WWWWWWWWWWWW

输出

复制20

20

示例2

输入

复制6 6 WWWWWW WEWS.W W.WK.W W.WD.W W.W..W WWWWWW

6 6
WWWWWW
WEWS.W
W.WK.W
W.WD.W
W.W..W
WWWWWW

输出

复制-1

-1

备注:

4 ≤ H, W≤ 500
'S', 'E', 'K', 'D'各出现恰好一次
迷宫的四周(最上面的一行、最下面的一行、最左边的一列以及最右边的一列) 都会是 'W'

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
char mp[505][505];
bool vis[505][505];
int n, m;
struct pos
{
    int x, y, step;
} s, e;
int bfs(int x1, int y1, int x2, int y2)
{
    int nx[][2] = {1, 0, 0, 1, 0, -1, -1, 0};
    s.x = x1, s.y = y1, e.x = x2, e.y = y2;
    queue<pos>Q;
    s.step = 0;
    Q.push(s);
    vis[s.x][s.y] = 1;
    while(!Q.empty())
    {
        pos t = Q.front();
        Q.pop();
        if(t.x == e.x && t.y == e.y)
            return t.step;
        for(int i = 0; i < 4; i++)
        {
            s.x = t.x + nx[i][0];
            s.y = t.y + nx[i][1];
            if(s.x >= 0 &&s.x < m && s.y >= 0 && s.y < n && mp[s.x][s.y] == '.'&& !vis[s.x][s.y])
            {
                vis[s.x][s.y] = 1;
                s.step = t.step + 1;
                Q.push(s);
            }
        }
    }
    return -1;
}
int main()
{
    scanf("%d%d", &m, &n);
    for(int i = 0; i < m; i++)
        scanf("%s", mp[i]);
    int sx, sy, ex, ey, kx, ky, dx, dy;
    for(int i = 0; i < m; i++)
        for(int j = 0; j < n; j++)
        {
            if(mp[i][j] == 'S') sx = i, sy = j;
            else if(mp[i][j] == 'E') ex = i, ey = j;
            else if(mp[i][j] == 'K') kx = i, ky = j;
            else if(mp[i][j] == 'D') dx = i, dy = j;
        }
    mp[sx][sy] = '.', mp[ex][ey] = '.', mp[kx][ky] = '.';
    int s = bfs(sx, sy, ex, ey);
    memset(vis, 0, sizeof vis);
    int s1 = bfs(sx, sy, kx, ky);
    memset(vis, 0, sizeof vis);
    mp[dx][dy] = '.';
    int s2 = bfs(kx, ky, dx, dy);
    memset(vis, 0, sizeof vis);
    int s3 = bfs(dx, dy, ex, ey);
    if(s == -1)
    {
        if(s1 == -1 || s2 == -1 || s3 == -1) puts("-1");
        else printf("%d\n", s1 + s2 + s3);
    }
    else printf("%d\n", min(s, (s1 + s2 + s3)));
    return 0;
}