NYOJ 82 - 迷宫寻宝（一）

描述

一个叫ACM的寻宝者找到了一个藏宝图，它根据藏宝图找到了一个迷宫，这是一个很特别的迷宫，迷宫里有N个编过号的门（N<=5)，它们分别被编号为A,B,C,D,E.为了找到宝藏，ACM必须打开门，但是，开门之前必须在迷宫里找到这个打开这个门所需的所有钥匙（每个门都至少有一把钥匙），例如：现在A门有三把钥匙，ACM就必须找全三把钥匙才能打开A门。现在请你编写一个程序来告诉ACM，他能不能顺利的得到宝藏。


输入
输入可能会有多组测试数据（不超过10组）。
每组测试数据的第一行包含了两个整数M,N(1<N,M<20)，分别代表了迷宫的行和列。接下来的M每行有N个字符，描述了迷宫的布局。其中每个字符的含义如下：
.表示可以走的路
S:表示ACM的出发点
G表示宝藏的位置
X表示这里有墙，ACM无法进入或者穿过。
A,B,C,D,E表示这里是门，a,b,c,d,e表示对应大写字母的门上的钥匙。
注意ACM只能在迷宫里向上下左右四个方向移动。

最后，输入0 0表示输入结束。

输出

每行输出一个YES表示ACM能找到宝藏，输出NO表示ACM找不到宝藏。

样例输入

4 4
S.X.
a.X.
..XG
....
3 4
S.Xa
.aXB
b.AG
0 0

样例输出

YES
NO

是一道搜索题，用BFS来做比较好，但是跟DFS较上劲了，本来想优雅的搜索搜索出来，但是处理钥匙和们的时候真是要了命了，于是把门拿出来单独处理了。

WA了好几次，蛋疼。

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

struct node
{
    int x;
    int y;
    int num;
}door[5];

char maps[25][25];
int max_key[5];//打开一个门需要的钥匙数
int find_key[5];//已经找到的钥匙数
bool flag;

void DFS(int i, int j);
void FromDoor();

void DFS(int i, int j)
{
    if(maps[i][j] != 'X')
    {
        if (maps[i][j] >= 'a' && maps[i][j] <= 'e')
                ++find_key[maps[i][j]-'a'];
        if (maps[i][j] >= 'A' && maps[i][j] <= 'E')
        {
            door[maps[i][j] - 'A'].x = i;
            door[maps[i][j] - 'A'].y = j;
            door[maps[i][j] - 'A'].num = 1;
            return;
        }
        if (maps[i][j] == 'G')
        {
            flag = true;
            return;
        }
        maps[i][j] = 'X';//标记为已经走过
        DFS(i-1, j);
        DFS(i+1, j);
        DFS(i, j-1);
        DFS(i, j+1);
        FromDoor();
    }
}

void FromDoor()//处理有门的地方
{
    for(int i = 0; i < 5; ++i)
    {
        if(door[i].num)
        {
            if(find_key[i] == max_key[i])//打开这个门所需的钥匙全部找到
            {
                maps[door[i].x][door[i].y] = 'X';
                DFS(door[i].x-1, door[i].y);
                DFS(door[i].x+1, door[i].y);
                DFS(door[i].x, door[i].y-1);
                DFS(door[i].x, door[i].y+1);
            }
        }
    }
}

int main()
{
    int tx, ty, m, n;
    while(cin >> m >> n && (m || n))
    {
        flag = false;
        memset(max_key, 0, sizeof(max_key));
        memset(find_key, 0, sizeof(find_key));
        memset(door, 0, sizeof(door));
        memset(maps, 'X', sizeof(maps));  //这样写也可以提交

        for(int i = 1; i <= m; ++i)
        {
            for(int j = 1; j <= n; ++j)
            {
                cin >> maps[i][j];
                if(maps[i][j] == 'S')
                {
                    tx = i;
                    ty = j;
                }
                else if(maps[i][j] >= 'a' && maps[i][j] <= 'e')
                    ++max_key[maps[i][j] - 'a'];
            }
        }
        DFS(tx, ty);
        if(flag)
            cout << "YES" << endl;
        else
            cout << "NO" << endl;
    }
    return 0;
}