E - A计划 HDU - 2102

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

思路：
用DFS深搜，最大才10*10，注意在搜索前进行判断而不是搜索到下一步后。

代码如下

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>

using namespace std;

int n,m,t,flag,vis1[12][12],vis2[12][12];  
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);  
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)  
        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]!='#')  
                    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]!='#')  
                    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=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;
}