首页 > 代码库 > NYOJ 82 迷宫寻宝(一)

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