【牛客练习赛12-B】迷宫(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
WE.W.S..W.KW
W..D..W....W
WWWWWWWWWWWW

输出

20

 

有两种情况：1.起点->终点 2.起点->钥匙->门->终点
注意第一种情况可以走钥匙的格子，不能走门的格子。

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