POJ 2157 Maze DFS

POJ 2157 Maze

炒鸡好的一道搜索题， 然而我并不会， 所以看得是 黄学长博客 才会做的。

题意：给你一张图， n， m。由字符组成，’S’ 是起点， ‘G’ 是终点， ‘X’ 代表墙， ‘A’, ‘B’, ‘C’, ‘D’, ‘E’ 是门， ‘a’, ‘b’, ‘c’, ‘d’, ‘e’ 是门的钥匙。其中钥匙数至少为1个，收集地图上全部 ‘a’ 才可能 开启 ‘A’， 其他也一样。输出能否从起点走到终点。

题解：收集完钥匙之后 看看对应的门是否走过， 如果走过，我们传送过去。代码易懂，有细节处理。Orz hzwer

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;

const int MAXN = 20 + 5;
const int dx[] = {0,0,1,0,-1};
const int dy[] = {0,1,0,-1,0};
bool h = false;
int n, m, tot = 0;
char maps[MAXN][MAXN];
bool vis_door[MAXN][MAXN], vis[MAXN][MAXN]; 

struct DOOR{
    int k, x, y;
}d[MAXN*MAXN];

struct KEY{
    int sum, tot;
}key[MAXN];

bool check(int x, int y)
{
    if(x <= 0 || x > n) return false;
    if(y <= 0 || y > m) return false;
    if(vis[x][y]) return false;
    if(maps[x][y] == 'X') return false;
    return true;
}

void dfs(int x, int y)
{
    vis[x][y] = 1;
    if(h) return;
    for(int i = 1; i <= 4; i ++)
    {
        int nx = x + dx[i];
        int ny = y + dy[i];
        if(!check(nx,ny))
            continue;
        if(maps[nx][ny] == 'G')
        {
            h = true;
            return;
        }
        else if(islower(maps[nx][ny]))
        {
            dfs(nx, ny);
            int k = (int)(maps[nx][ny]-'a');
            key[k].sum ++;
            if(key[k].sum == key[k].tot)
                for(int j = 1; j <= tot; j ++)
                    if(d[j].k == k && vis_door[d[j].x][d[j].y])
                        dfs(d[j].x, d[j].y);
        }
        else if(isupper(maps[nx][ny])) // 不可能为 S || G 
        {
            vis_door[nx][ny] = 1;
            int k = (int)(maps[nx][ny]-'A');
            if(key[k].sum && key[k].sum == key[k].tot)
                dfs(nx, ny);
        }
        else dfs(nx,ny);
    }
}

int main()
{
    while(cin >> n >> m && n && m)
    {
        memset(vis_door, 0, sizeof(vis_door));
        memset(vis, 0, sizeof(vis));
        memset(d, 0, sizeof(d));
        memset(key, 0, sizeof(key));
        h = false; tot = 0;
        int sx, sy;
        for(int i = 1; i <= n; i ++)
        {
            scanf("%s", maps[i]+1);
            for(int j = 1; j <= m; j ++)
            {
                if(maps[i][j] == 'S')
                    sx = i, sy = j;
                else if(islower(maps[i][j]))
                {
                    int k = (int)(maps[i][j]-'a');
                    key[k].tot ++;
                }
                else if(isupper(maps[i][j])) //
                {
                    if(maps[i][j] == 'G' || maps[i][j] == 'S')// 可不加这一行
                        continue;
                    int k = (int)(maps[i][j]-'A');
                    tot ++;
                    d[tot].x = i;
                    d[tot].y = j;
                    d[tot].k = k;
                }
            }
        }

        dfs(sx, sy);

        if(h) puts("YES");
        else puts("NO");
    }
    return 0;
}