HDU 2102 A计划

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<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = 15;
int n, m, T, C, a[2][maxn][maxn], t[2][maxn][maxn], f[2][maxn][maxn];
int b[4] = { 0, 0, 1, -1 };
int c[4] = { 1, -1, 0, 0 };
char s[maxn];

struct abc
{
	int x, y, z;
	abc(){}
	abc(int x, int y, int z) :x(x), y(y), z(z){}
};

void bfs(int x, int y, int z)
{
	memset(f, 0, sizeof(f));
	memset(t, -1, sizeof(t));
	queue<abc> p;
	abc q;
	p.push(abc(x, y, z));
	t[0][1][1] = 0;
	f[x][y][z] = 1;
	int flag = 0;
	while (!p.empty())
	{
		q = p.front(); p.pop();
		if (a[q.x][q.y][q.z] >= 2)
		{
			if (a[q.x][q.y][q.z] >= 3) { flag = 1; break; }
			if (!f[1 - q.x][q.y][q.z] && a[1 - q.x][q.y][q.z])
			{
				f[1 - q.x][q.y][q.z] = 1;
				t[1 - q.x][q.y][q.z] = t[q.x][q.y][q.z];
				p.push(abc(1 - q.x, q.y, q.z));
			}
		}
		else 
		for (int i = 0; i<4; i++)
			if (q.y + b[i]>0 && q.y + b[i] <= n)
				if (q.z + c[i]>0 && q.z + c[i] <= m)
					if (a[q.x][q.y + b[i]][q.z + c[i]])
						if (!f[q.x][q.y + b[i]][q.z + c[i]])
						{
							f[q.x][q.y + b[i]][q.z + c[i]] = 1;
							t[q.x][q.y + b[i]][q.z + c[i]] = t[q.x][q.y][q.z] + 1;
							p.push(abc(q.x, q.y + b[i], q.z + c[i]));
						}
	}
	if (flag&&t[q.x][q.y][q.z] <= T) printf("YES\n");
	else printf("NO\n");
}

int main()
{
	scanf("%d", &C);
	while (C--)
	{
		scanf("%d%d%d", &n, &m, &T);
		for (int i = 0; i < n + n; i++)
		{
			scanf("%s", s);
			for (int j = 1; j <= m; j++)
			{
				switch (s[j - 1]){
					case 'S':a[i / n][i % n + 1][j] = 1; break;
					case '.':a[i / n][i % n + 1][j] = 1; break;
					case '*':a[i / n][i % n + 1][j] = 0; break;
					case '#':a[i / n][i % n + 1][j] = 2; break;
					case 'P':a[i / n][i % n + 1][j] = 3; break;
				}
			}
		}
		bfs(0, 1, 1);
	}
	return 0;
}