Description
可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后,而今,不幸的她再一次面临生命的考验。魔王已经发出消息说将在T时刻吃掉公主,因为他听信谣言说吃公主的肉也能长生不老。年迈的国王正是心急如焚,告招天下勇士来拯救公主。不过公主早已习以为常,她深信智勇的骑士LJ肯定能将她救出。
现据密探所报,公主被关在一个两层的迷宫里,迷宫的入口是S(0,0,0),公主的位置用P表示,时空传输机用#表示,墙用*表示,平地用.表示。骑士们一进入时空传输机就会被转到另一层的相对位置,但如果被转到的位置是墙的话,那骑士们就会被撞死。骑士们在一层中只能前后左右移动,每移动一格花1时刻。层间的移动只能通过时空传输机,且不需要任何时间。
现据密探所报,公主被关在一个两层的迷宫里,迷宫的入口是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
刚看到这道题的时候,我就想到用DFS深搜,毕竟最大才10*10,结果TLE了几次。
这道题有几个坑点。
题目上说的是骑士能不能在T时刻找到公主,然而并不是一定要在T时刻,在T时刻之前找到也行,另外,如果第一层和第二层同一位置都是#那么就会一直传送,自然就不能找到公主了。
还有就是DFS回溯很费时间,尽量在搜索前进行判断而不是搜索到下一步后。
具体代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
int n,m,t,flag,vis1[12][12],vis2[12][12]; //flag标记是否到达终点,vis1和vis2分别标记第一层和第二层走过的地方
int xi[4]= {0,0,1,-1},yi[4]= {1,-1,0,0}; // 用数组存储四个方向
int rx,ry,num;
char a[12][12],b[12][12];
void map1(int x,int y,int step); // step为花费的时间
void map2(int x,int y,int step);
int h(int x,int y)
{
return abs(x-rx)+abs(y-ry); // 当前位置距终点的距离,也就是需要的时间
}
void map2(int x,int y,int step)
{
if(flag||step>t)
return ;
if(step+h(x,y)>t&&num==2) //num表示终点在地图的哪一层
return ;
if(b[x][y]=='P')
{
if(step<=t)
flag=1;
return ;
}
int xx,yy;
for(int i=0; i<4; i++)
{
xx=x+xi[i];
yy=y+yi[i];
if(xx>=0&&yy>=0&&xx<m&&yy<n&&b[xx][yy]!='*'&&!vis2[xx][yy])
{
vis2[xx][yy]=1;
if(b[xx][yy]=='#')
{
if(a[xx][yy]!='*'&&a[xx][yy]!='#') //在DFS前判断能不能走到下一层
map1(xx,yy,step+1);
}
else map2(xx,yy,step+1);
vis2[xx][yy]=0;
}
}
return ;
}
void map1(int x,int y,int step)
{
if(flag||step>t)
return ;
if(step+h(x,y)>t&&num==1)
return ;
if(a[x][y]=='P')
{
if(step<=t)
flag=1;
return ;
}
int xx,yy;
for(int i=0; i<4; i++)
{
xx=x+xi[i];
yy=y+yi[i];
if(xx>=0&&yy>=0&&xx<m&&yy<n&&a[xx][yy]!='*'&&!vis1[xx][yy])
{
vis1[xx][yy]=1;
if(a[xx][yy]=='#')
{
if(b[xx][yy]!='*'&&b[xx][yy]!='#') //在DFS前判断能不能走到下一层
map2(xx,yy,step+1);
}
else map1(xx,yy,step+1);
vis1[xx][yy]=0;
}
}
return ;
}
int main()
{
int c;
scanf("%d",&c);
while(c--)
{
scanf("%d %d %d",&m,&n,&t);
flag=num=0;
memset(vis1,0,sizeof(vis1));
memset(vis2,0,sizeof(vis2));
for(int i=0; i<m; i++)
{
scanf("%s",a[i]);
for(int j=0; j<n; j++)
if(a[i][j]=='P') // rx,ry标记终点坐标,num标记终点在哪一层
{
rx=i;
ry=j;
num=1;
}
}
for(int i=0; i<m; i++)
{
scanf("%s",b[i]);
for(int j=0; j<n; j++)
if(b[i][j]=='P')
{
rx=i;
ry=j;
num=2;
}
}
vis1[0][0]=1;
map1(0,0,0);
if(flag)
printf("YES\n");
else printf("NO\n");
}
return 0;
}