hdu 2102 搜索

A计划

Time Limit: 3000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 11187 Accepted Submission(s): 2728

Problem 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

题目不难，但是有点坑，因为有个地方忘了被坑去了5次wa，哭晕在厕所。

#include<iostream>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
char map[2][15][15];
int mark[2][15][15];
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
struct Fly
{
    int i,j,k;
}fly[450];
struct mi
{
    int i,j,k,time;
};
int cas;
int n,m,t;
void init()
{
    int sum=0;
    cin>>n>>m>>t;
    for(int i=0;i<2;i++)
    {
        for(int j=0;j<n;j++)
        {
            for(int k=0;k<m;k++)
            {
                cin>>map[i][j][k];
                if(map[i][j][k]=='#')
                {
                    fly[sum].i=i;
                    fly[sum].j=j;
                    fly[sum].k=k;
                    sum++;
                }
            }
        }
    }
    for(int i=0;i<sum;i++)
    {
        if(map[1-fly[i].i][fly[i].j][fly[i].k]=='*'||map[1-fly[i].i][fly[i].j][fly[i].k]=='#')
            map[fly[i].i][fly[i].j][fly[i].k]=map[1-fly[i].i][fly[i].j][fly[i].k]='*';
    }
}
void search()
{
    memset(mark,0,sizeof(mark));
    mi temp,temp2;
    temp.i=temp.j=temp.k=temp.time=0;
    mark[0][0][0]=1;
    queue<mi>q;
    q.push(temp);
    while(!q.empty())
    {
        temp=q.front();
        q.pop();
        if(temp.time==t)
        {
            cout<<"NO"<<endl;
            return;
        }
        temp2.time=temp.time+1;
        for(int i=0;i<4;i++)
        {
            temp2.i=temp.i;
            temp2.j=temp.j+dx[i];
            temp2.k=temp.k+dy[i];
            if(temp2.j>=0&&temp2.j<n&&temp2.k>=0&&temp2.k<m&&map[temp2.i][temp2.j][temp2.k]!='*'&&!mark[temp2.i][temp2.j][temp2.k])
            {
                mark[temp2.i][temp2.j][temp2.k]=1;
                if(map[temp2.i][temp2.j][temp2.k]=='#')
                {
                    temp2.i=1-temp2.i;
                    if(mark[temp2.i][temp2.j][temp2.k])continue;
                    else mark[temp2.i][temp2.j][temp2.k]=1;
                }
                if(map[temp2.i][temp2.j][temp2.k]=='P')
                {
                    cout<<"YES"<<endl;
                    return;
                }
                q.push(temp2);
            }
        }
    }
    cout<<"NO"<<endl;
}
int main()
{
    cin>>cas;
    while(cas--)
    {
        init();
        search();
    }
    return 0;
}