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

 1 #include <cstdio>

 2 #include <cstring>

 3 #include <set>

 4 #define LL long long

 5 using namespace std;

 6 char table[23][23];

 7 int rows,cols,totKey[5],hasKey[5],sx,sy;

 8 set<LL>vis;

 9 bool isVis(int x,int y){

10     LL temp = 0;

11     for(int i = 0; i < 5; i++)

12         temp = (temp<<9)+hasKey[i];

13     temp = (temp<<10)+(x<<5)+y;

14     if(vis.count(temp)) return true;

15     vis.insert(temp);

16     return false;

17 }

18 bool dfs(int x,int y) {

19     if(table[x][y] == 'X') return false;

20     if(table[x][y] == 'G')

21         return true;

22     if(isVis(x,y)) return false;

23     static  int dir[4][2] = {0,-1,0,1,-1,0,1,0};

24     if(table[x][y] >= 'a' && table[x][y] <= 'e') {

25         hasKey[table[x][y]-'a']++;

26         table[x][y] = '.';

27     } else if(table[x][y] >= 'A' && table[x][y] <= 'E') {

28         if(hasKey[table[x][y]-'A'] < totKey[table[x][y]-'A'])

29             return false;

30     }

31     char ch = table[x][y];

32     table[x][y] = '.';

33     for(int i = 0; i < 4; i++)

34         if(dfs(x+dir[i][0],y+dir[i][1])) return true;

35     if(table[x][y] != '.') table[x][y] = ch;

36     return false;

37 }

38 int main() {

39     int i,j;

40     while(scanf("%d %d",&rows,&cols),rows||cols) {

41         memset(totKey,0,sizeof(totKey));

42         memset(hasKey,0,sizeof(hasKey));

43         memset(table,'X',sizeof(table));

44         for(i = 1; i <= rows; i++) {

45             scanf("%s",table[i]+1);

46             table[i][strlen(table[i])]='X';

47             for(j = 1; j <= cols; j++) {

48                 if(table[i][j] == 'S') {

49                     table[sx = i][sy = j] = '.';

50                 } else if(table[i][j] >= 'a' && table[i][j] <= 'e') {

51                     totKey[table[i][j]-'a']++;

52                 }

53             }

54         }

55         vis.clear();

56         for(i = 0; i < 5; i++)

57             if(!totKey[i]) totKey[i] = 500;

58         printf("%s\n",dfs(sx,sy)?"YES":"NO");

59     }

60     return 0;

61 }

View Code