hdu 2102：A计划（BFS）

http://acm.hdu.edu.cn/showproblem.php?pid=2102

Problem Description

可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在T时刻吃掉公主，因为他听信谣言说吃公主的肉也能长生不老。年迈的国王正是心急如焚，告招天下勇士来拯救公主。不过公主早已习以为常，她深信智勇的骑士LJ肯定能将她救出。
现据密探所报，公主被关在一个两层的迷宫里，迷宫的入口是S（0，0，0），公主的位置用P表示，时空传输机用#表示，墙用*表示，平地用.表示。骑士们一进入时空传输机就会被转到另一层的相对位置，但如果被转到的位置是墙的话，那骑士们就会被撞死。骑士们在一层中只能前后左右移动，每移动一格花1时刻。层间的移动只能通过时空传输机，且不需要任何时间。

Input

输入的第一行C表示共有C个测试数据，每个测试数据的前一行有三个整数N，M，T。 N，M迷宫的大小N*M（1 <= N,M <=10)。T如上所意。接下去的前N*M表示迷宫的第一层的布置情况，后N*M表示迷宫第二层的布置情况。

Output

如果骑士们能够在T时刻能找到公主就输出“YES”，否则输出“NO”。

Sample Input

1
5 5 14
S*#*.
.#...
.....
****.
...#.

..*.P
#.*..
***..
...*.
*.#..

Sample Output

YES

解题思路：

记录步数，如果是#就改变层数，注意传送后就是公主的情况。

#include <stdio.h>
#include <string.h>
#include <queue>
#define N 20
using namespace std;
char str[2][N][N];
bool book[2][N][N];
int n, m, t;
struct date{
	int x;
	int y;
	int z;
	int temp;
};
int bfs(int x, int y, int z)
{
	int nex[4][2]={0,1, 0,-1, 1,0, -1,0};
	int ty, tz;
	queue<date>q;	
	q.push(date{x, y, z, 0});
	memset(book, false, sizeof(book));
	book[x][y][z]=true;
	while(!q.empty())
	{
		date u=q.front();
		q.pop();
		if(str[u.x][u.y][u.z]=='P')
		{
			if(u.temp<=t)
				return 1;
			return 0;
		}
		
		if((str[u.x][u.y][u.z]=='#') && (str[u.x^1][u.y][u.z]=='*' || str[u.x^1][u.y][u.z]=='#'))
			continue;
		if(str[u.x][u.y][u.z]=='#')
			u.x=u.x^1;
		if(str[u.x][u.y][u.z]=='P')
		{
			if(u.temp<=t)
				return 1;
			return 0;
		}
		for(int i=0; i<4; i++)
		{
			ty=u.y+nex[i][0];
			tz=u.z+nex[i][1];
			if(ty>=n || tz>=m || ty<0 || tz<0 || str[u.x][ty][tz]=='*')
				continue;
		
			if(book[u.x][ty][tz]==false)
			{
				book[u.x][ty][tz]=true;
				q.push(date{u.x, ty, tz, u.temp+1});
				
			}
		}
	}
	return 0;
}
int main()
{
	int T;
	scanf("%d", &T);
	while(T--)
	{
		scanf("%d%d%d", &n, &m, &t);
		for(int i=0; i<2; i++)
			for(int j=0; j<n; j++)
				scanf("%s", str[i][j]);
		if(bfs(0, 0, 0))
			printf("YES
");
		else
			printf("NO
");
	}
	return 0;
}