kuangbin专题二 搜索进阶 I - A计划

I - A计划

  

可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在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

这次很快就写完了，wa了几次，忘记判断两边都是传送门的时候了，自己检查出来了。。。

#include<iostream>
#include<string.h>
#include<queue>
#include<string>
using namespace std ;
typedef long long ll;
typedef unsigned long long ull;
const int _max =15;
char mp[2][_max][_max];
int dx[]={1,-1,0,0};
int dy[]={0,0,1,-1};
bool vis[2][_max][_max];
struct node{
	int x,y;
	bool z;
	int t;
};
int n,m,T,x,y,z;
bool check( node s ){
	if(!vis[s.z][s.x][s.y]) return false;
	if(mp[s.z][s.x][s.y]=='*')return false;
	if(s.x>n-1||s.x<0||s.y>m-1||s.y<0) return false;
	return true;
}
void bfs(){
	queue<node>q;
	node now,next;
	now.x=x;
	now.y=y;
	now.z=z;
	now.t=0;
	q.push(now);
	vis[now.z][now.x][now.y]=0;
	while(!q.empty()){
		now=q.front();
		q.pop();
		if(mp[now.z][now.x][now.y]=='P'&&now.t<=T){
			cout<<"YES"<<endl;
			return;
		}
		for(int i=0;i<4;i++){
			next.x=now.x+dx[i];
			next.y=now.y+dy[i];
			next.z=now.z;
			next.t=now.t+1;
            if(!check(next)) continue;
			if(mp[next.z][next.x][next.y]=='#')
			{
				next.z=!next.z;
				if(mp[next.z][next.x][next.y]=='*'||vis[next.z][next.x][next.y]==0||mp[next.z][next.x][next.y]=='#') {
					vis[next.z][next.x][next.y]=vis[now.z][next.x][next.y]=0;
					continue;
				}
				vis[next.z][next.x][next.y]=vis[now.z][next.x][next.y]=0;
				q.push(next);
			}
			else {
				vis[next.z][next.x][next.y]=0;
				q.push(next);
			}
		}
		}
		cout<<"NO"<<endl; 
}
int main(){
	int c;
	cin>>c;
	while(c--){
	cin>>n>>m>>T;
	memset(vis,1,sizeof(vis));
	for(int k = 0 ; k <2 ; k++)
		for(int i=0;i<n;i++)
		   for( int j =0; j < m ; j++){
		   	cin>>mp[k][i][j];
		   	if(mp[k][i][j]=='S'){
		   		x=i;
				y=j;
				z=k; 
		   	}
			   }
			   bfs();
	}
	return 0;
}