nyoj 82 搜索

链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=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

#include<iostream>#include<cstring>#include<cstdio>#include<queue>#include<cmath>#include<algorithm>#include<cstdlib>#include<vector>using namespace std;int p = 1;#define CLR(arr, what) memset(arr, what, sizeof(arr))const int N = 25, M = 5;const char dir[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}}; struct node{int x, y;}src, des, doorpos[5];      char maze[N][N];     bool visit[N][N];     bool door[5];      int key[5];         int row, col;queue<node> q;bool travel(node cur) {if(cur.x < 0 || cur.y < 0 || cur.x >= row || cur.y >= col || maze[cur.x][cur.y] == 'X')return false;return true;}bool check(){for(int i = 0; i < 5; ++i)if(key[i] != 0)return false;return true;}bool BFS(){q.push(src);visit[src.x][src.y] = true;while(!q.empty()){node cur, temp;cur = q.front();q.pop();visit[src.x][src.y] = false;if(maze[cur.x][cur.y] >= 'a' && maze[cur.x][cur.y] <= 'e' && key[maze[cur.x][cur.y] - 'a'] > 0){key[maze[cur.x][cur.y] - 'a']--;if(key[maze[cur.x][cur.y] - 'a'] == 0 && door[maze[cur.x][cur.y] - 'a'] == true){node in;in.x = doorpos[maze[cur.x][cur.y] - 'a'].x;in.y = doorpos[maze[cur.x][cur.y] - 'a'].y;if(maze[in.x][in.y] == 'X')q.push(in);}}if(maze[cur.x][cur.y] >= 'A' && maze[cur.x][cur.y] <= 'E'){if(key[maze[cur.x][cur.y] - 'A'] > 0){maze[cur.x][cur.y] = 'X';continue;}}maze[cur.x][cur.y] = 'X';if(cur.x == des.x && cur.y == des.y && check()) return true;for(int i = 0; i < 4; ++i) {temp.x = cur.x + dir[i][0]; temp.y = cur.y + dir[i][1];if(travel(temp) == false || visit[temp.x][temp.y])continue;q.push(temp);visit[temp.x][temp.y] = true;}}return false;}int main(){while(scanf("%d%d", &row, &col) && (row + col)){CLR(door, false); CLR(key, 0); CLR(visit, false);for(int i = 0; i < 5; ++i){doorpos[i].x = -1; doorpos[i].y = -1;}while(!q.empty())q.pop();for(int i = 0; i < row; ++i) //迷宫处理{scanf("%s", maze[i]);for(int j = 0; j < col; ++j){if(maze[i][j] == 'S')src.x = i, src.y = j;else if(maze[i][j] == 'G')des.x = i, des.y = j;else if(maze[i][j] >= 'A' && maze[i][j] <= 'E'){door[maze[i][j] - 'A'] = true;doorpos[maze[i][j] - 'A'].x = i;doorpos[maze[i][j] - 'A'].y = j;}else if(maze[i][j] >= 'a' && maze[i][j] <= 'e')key[maze[i][j] - 'a']++;}}printf(BFS() ? "YES\n" : "NO\n");}return 0;}