【nowcoder 练习赛12】迷宫（bfs）

链接：https://www.nowcoder.net/acm/contest/68/B
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld
题目描述
这是一个关于二维迷宫的题目。我们要从迷宫的起点 ‘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
输出
20
示例2
输入
6 6
WWWWWW
WEWS.W
W.WK.W
W.WD.W
W.W..W
WWWWWW
输出
-1
备注:
4 ≤ H, W≤ 500
‘S’, ‘E’, ‘K’, ‘D’各出现恰好一次
迷宫的四周(最上面的一行、最下面的一行、最左边的一列以及最右边的一列) 都会是 ‘W’

分析：就是一个简单的bfs，lg卡在了83.33%，
其实只需要 三次bfs就可以。 一次bfs：找出起点到终点的距离tmp1， 两次bfs找出起点到钥匙的距离tmp2，钥匙到终点的距离tmp3。 取其中较小者，判断是否INF

【bfs】
代码：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <map>
#include <set>
#include <bitset>
#include <cctype>
#include <cstdlib>
#include <queue>
#include <cmath>
#include <stack>
#include <ctime>
#include <string>
#include <vector>
#include <sstream>
#include <functional>
#include <algorithm>
using namespace std;

#define mem(a,n) memset(a,n,sizeof(a))
#define memc(a,b) memcpy(a,b,sizeof(b))
#define rep(i,a,n) for(int i=a;i<n;i++) ///[a,n)
#define dec(i,n,a) for(int i=n;i>=a;i--)///[n,a]
#define pb push_back
#define IO ios::sync_with_stdio(false)
#define fre freopen("in.txt","r",stdin)
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
typedef unsigned long long ull;
const double PI=acos(-1.0);
const double eps=1e-8;
const double e=2.7182818;
const int INF=0x3f3f3f3f;
const int MOD=998244353;
const int N=5e2+5;
const ll maxn=5e4;
const int dir[4][2]= {-1,0,1,0,0,-1,0,1};
int h,w;
char a[N][N];
bool vis[N][N];
struct Node
{
    int x,y,cnt;
    Node(int x,int y,int cnt):x(x),y(y),cnt(cnt) {}
};
bool in(int x,int y)///判断坐标（x,y）是否合法
{
    if(x>=0&&x<h&&y>=0&&y<w) return true;
    return false;
}
int bfs(int x,int y,int tx,int ty)
{
    rep(i,0,h+1) rep(j,0,w+1) vis[i][j]=0;
    queue<Node>que;
    Node now=Node(x,y,0);
    vis[x][y]=1;
    que.push(now);
    while(!que.empty())
    {
        now=que.front();
        que.pop();
        if(now.x==tx&&now.y==ty) return now.cnt;
        rep(i,0,4)
        {
            int nx=now.x+dir[i][0];
            int ny=now.y+dir[i][1];
            if(in(nx,ny)&&!vis[nx][ny]&&a[nx][ny]!='W')
            {
                vis[nx][ny]=1;
                que.push(Node(nx,ny,now.cnt+1));
            }
        }
    }
    return INF;
}
int main()
{
   // fre;
    IO;
    while(cin>>h>>w)
    {
        int sx,sy,ex,ey;
        int kx,ky,dx,dy;
        rep(i,0,h)
        {
            cin>>a[i];
            rep(j,0,w)
            {
                if(a[i][j]=='S') sx=i,sy=j;
                if(a[i][j]=='E') ex=i,ey=j;
                if(a[i][j]=='D') dx=i,dy=j,a[i][j]='W';
                if(a[i][j]=='K') kx=i,ky=j;
            }
        }
        int tmp1=bfs(sx,sy,ex,ey);
        int tmp2=bfs(sx,sy,kx,ky);
        a[dx][dy]='.';
        int tmp3=bfs(kx,ky,ex,ey);
        int ans=min(tmp1,tmp2+tmp3);
        if(ans==INF) cout<<-1<<endl;
        else cout<<ans<<endl;
    }
    return 0;
}