迷宫问题进阶

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.KW

W. .D..W....W

WW WWWWWWWWWW

输出：

20

代码：

/*
    思路：先看不经过门能不能到达终点，并求出能到达终点距离；
如果经过门也能达到终点，先求出到达钥匙的距离再求钥匙到达终点的距离，相加便是经过门到达终点的距离；

最后比较两种走法哪个距离最短；

自己写的代码太复杂最后也wa了，参考了别人的代码。
*/
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
struct node
{
   int x,y,step;
}ss,ee,kk;
char g[505][505];
bool vis[505][505];
int dx[4] = {0, 0, -1, 1};
int dy[4] = {-1, 1, 0, 0};
int n,m,ans;
void bfs(node s, node e, bool key)//key表示是否有钥匙
{
    memset(vis, 0, sizeof(vis));
    queue<node> q;
    node next;
    next.x = s.x;
    next.y = s.y;
    vis[s.x][s.y] = 1;
    next.step = s.step;
    q.push(next);
    while(!q.empty())
    {
        next = q.front();
        //printf("%d %d %d\n",next.x,next.y,next.step);
        q.pop();
        for(int i = 0; i < 4; i++)
        {
            int x = next.x + dx[i];
            int y = next.y + dy[i];
            if(x > 0 && x < n-1 && y > 0 && y < m-1 && !vis[x][y]
               && g[x][y] != 'W' && (g[x][y] != 'D' || (g[x][y] == 'D' && key)))
            {
                if(g[x][y] == 'E')
                {
                    ans = min(ans,next.step+1);
                    return ;
                }
                if(g[x][y] == 'K')
                    kk.step = next.step+1;
                vis[x][y] = 1;
                node temp;
                temp.x = x;
                temp.y = y;
                temp.step = next.step + 1;
                q.push(temp);
            }
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 0; i < n; i++)
        scanf("%s",g[i]);
    for(int i = 0; i < n; i++)
        for(int j = 0; j < m; j++)
    {
        if(g[i][j] == 'S')
            ss.x = i,ss.y = j,ss.step = 0;
        if(g[i][j]=='E')
            ee.x=i,ee.y=j;
        if(g[i][j]=='K')
            kk.x=i,kk.y=j,kk.step=-1;
    }
    ans = inf;
    bfs(ss,ee,0);//没有钥匙
    if(kk.step != -1)
        bfs(kk,ee,1);//有钥匙
    if(ans == inf)
        printf("-1\n");
    else
        printf("%d\n",ans);
    return 0;
}