NYOJ 82 迷宫寻宝（一）

迷宫寻宝（一）

时间限制：1000 ms  |  内存限制：65535 KB

难度：4

 

描述

一个叫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

来源

POJ月赛改编

上传者

张云聪

解题：压缩+dfs

#include <cstdio>
#include <cstring>
#include <set>
#define LL long long
using namespace std;
char table[23][23];
int rows,cols,totKey[5],hasKey[5],sx,sy;
set<LL>vis;
bool isVis(int x,int y){
    LL temp = 0;
    for(int i = 0; i < 5; i++)
        temp = (temp<<9)+hasKey[i];
    temp = (temp<<10)+(x<<5)+y;
    if(vis.count(temp)) return true;
    vis.insert(temp);
    return false;
}
bool dfs(int x,int y) {
    if(table[x][y] == 'X') return false;
    if(table[x][y] == 'G')
        return true;
    if(isVis(x,y)) return false;
    static  int dir[4][2] = {0,-1,0,1,-1,0,1,0};
    if(table[x][y] >= 'a' && table[x][y] <= 'e') {
        hasKey[table[x][y]-'a']++;
        table[x][y] = '.';
    } else if(table[x][y] >= 'A' && table[x][y] <= 'E') {
        if(hasKey[table[x][y]-'A'] < totKey[table[x][y]-'A'])
            return false;
    }
    char ch = table[x][y];
    table[x][y] = '.';
    for(int i = 0; i < 4; i++)
        if(dfs(x+dir[i][0],y+dir[i][1])) return true;
    if(table[x][y] != '.') table[x][y] = ch;
    return false;
}
int main() {
    int i,j;
    while(scanf("%d %d",&rows,&cols),rows||cols) {
        memset(totKey,0,sizeof(totKey));
        memset(hasKey,0,sizeof(hasKey));
        memset(table,'X',sizeof(table));
        for(i = 1; i <= rows; i++) {
            scanf("%s",table[i]+1);
            table[i][strlen(table[i])]='X';
            for(j = 1; j <= cols; j++) {
                if(table[i][j] == 'S') {
                    table[sx = i][sy = j] = '.';
                } else if(table[i][j] >= 'a' && table[i][j] <= 'e') {
                    totKey[table[i][j]-'a']++;
                }
            }
        }
        vis.clear();
        for(i = 0; i < 5; i++)
            if(!totKey[i]) totKey[i] = 500;
        printf("%s
",dfs(sx,sy)?"YES":"NO");
    }
    return 0;
}