HDU 2102 A计划（深搜）

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 
    
    

 

刚看到这道题的时候，我就想到用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;
}