HDU 2102 A计划

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

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

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

Sample Input
1
5 5 14
S*#*.
.#…
…
****.
…#.

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

Sample Output
YES

刚看到这道题有点懵，来了个三维数组，但感觉跟一般的差不多，就直接就动手用bfs，后来代码出来了，但一直WA，后来才发现忘了考虑当坐时空机转到另一层对应位置时，如果是墙或者同样也是时空机，那么骑士就一定无法到达指定位置，因此我先找出符合这种情况的所有位置，直接把这两层对应位置都换成’ * ’ , 这样问题就容易解决多了。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
char s[2][11][11];
int vis[2][11][11];
int dx[4]={1,0,-1,0};
int dy[4]={0,1,0,-1};
int n,m,t,flag,sk,sx,sy;
struct node
{
	int k,x,y,st;
};
bool judge(int k,int x,int y)
{
	if(x<0||x>=n||y<0||y>=m||s[k][x][y]=='*'||vis[k][x][y])
		return false;
	return true;
}
void bfs()
{
	node temp,tmp;
	queue<node> q;
	temp.x=sx;temp.y=sy;temp.k=sk;temp.st=0;
	vis[sk][sx][sy]=1;
	q.push(temp);
	while(!q.empty())
	{
		tmp=q.front();
		q.pop();
		if(tmp.st>t) return ;
		if(s[tmp.k][tmp.x][tmp.y]=='P'){
			flag=1;
			//cout<<tmp.st<<endl;
			return ;
		}
		for(int i=0;i<4;i++){
			int xx=tmp.x+dx[i];
			int yy=tmp.y+dy[i];
			int kk=tmp.k;
			int t=tmp.st+1;
			if(judge(kk,xx,yy)){
				temp.k=kk;
				temp.x=xx;
				temp.y=yy;				
				if(s[kk][xx][yy]=='#'){
					temp.k=1-tmp.k;
					temp.st=t;
					vis[temp.k][xx][yy]=1;
					vis[kk][xx][yy]=1;
					q.push(temp);
				}
				else{
					temp.st=t;
					vis[kk][xx][yy]=1;
					q.push(temp);
				}
			}
		}
	}
}
int main()
{
	int c;
	cin>>c;
	while(c--)
	{
		cin>>n>>m>>t;
		for(int i=0;i<2;i++){
			for(int j=0;j<n;j++){
				scanf("%s",&s[i][j]);
			}
		}
		for(int i=0;i<2;i++){
			for(int j=0;j<n;j++){
				for(int d=0;d<m;d++){
					if(s[i][j][d]=='S'){
						sk=i;sx=j;sy=d;
					}
					else if(s[i][j][d]=='#'&&s[1-i][j][d]=='#'){
						s[i][j][d]='*';
						s[1-i][j][d]='*';
					}else if(s[i][j][d]=='#'&&s[1-i][j][d]=='*'){
						s[i][j][d]='*';
					}
				}
			}
		}
		flag=0;
		memset(vis,0,sizeof(vis));
		bfs();
		if(flag)
			cout<<"YES"<<endl;
		else
			cout<<"NO"<<endl;
	}
	return 0;
}