首页 > 代码库 > HDU-2102-A计划(BFS+优先队列)

HDU-2102-A计划(BFS+优先队列)

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
 

Source
HDU 2007-6 Programming Contest 


思路:水BFS,被坑了好久。。。注意上下两层都是传送机的情况。


#include <cstdio>
#include <queue>
using namespace std;

struct S{
int x,y,z,step;

bool operator<(const S &p) const
{
    return step>p.step;
}
}t;

char mp[2][10][11];
bool vis[2][10][10];
int nxt[4][2]={{0,1},{1,0},{0,-1},{-1,0}};

int main()
{
    int T,n,m,lim,i,j,k;

    scanf("%d",&T);

    while(T--)
    {
        scanf("%d%d%d",&n,&m,&lim);

        for(i=0;i<2;i++) for(j=0;j<n;j++) scanf("%s",mp[i][j]);

        for(i=0;i<2;i++) for(j=0;j<n;j++) for(k=0;k<m;k++) vis[i][j][k]=0;

        int top=0,bottom=1;

        vis[0][0][0]=1;
        mp[0][0][0]='.';

        t.x=0;
        t.y=0;
        t.z=0;
        t.step=0;

        priority_queue<S>que;

        que.push(t);

        while(!que.empty())
        {
            t=que.top();

            if(mp[t.x][t.y][t.z]=='P')
            {
                puts("YES");

                break;
            }

            que.pop();

            for(i=0;i<4;i++)
            {
                t.y+=nxt[i][0];
                t.z+=nxt[i][1];

                if(t.y>=0 && t.y<n && t.z>=0 && t.z<m && !vis[t.x][t.y][t.z] && t.step<lim)
                {
                    vis[t.x][t.y][t.z]=1;

                    if(mp[t.x][t.y][t.z]=='.' || mp[t.x][t.y][t.z]=='P')
                    {
                        t.step++;

                        que.push(t);

                        t.step--;
                    }
                    else if(mp[t.x][t.y][t.z]=='#')
                    {
                        if(!vis[!t.x][t.y][t.z] && (mp[!t.x][t.y][t.z]=='.' || mp[!t.x][t.y][t.z]=='P'))
                        {
                            vis[!t.x][t.y][t.z]=1;

                            t.x=!t.x;

                            t.step++;

                            que.push(t);

                            t.step--;

                            t.x=!t.x;
                        }
                    }
                }

                t.y-=nxt[i][0];
                t.z-=nxt[i][1];
            }
        }

        if(que.empty()) puts("NO");
    }
}