HDOJ-2102A计划(简单BFS）

A计划

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

---

思路：这题属于简单的BFS，需要特别注意几个陷阱：

1.传输机的另一头如果是传输机，那么就相当于一个死循环。
2.传输机另一头有可能就是公主，因此判断的时候要注意。

下面附上AC代码：

#include<bits/stdc++.h>
using namespace std;
#define M(a, b) memset(a, b, sizeof(a))

char maze[2][15][15];
bool vis[2][15][15];
int T, n, m, sx, sy, sf, dir[4][2] = {0,1, 0,-1, 1,0, -1,0};
typedef struct{
    int f;
    int x, y;
    int t;
}node;

bool ok(int ff, int xx, int yy){
    if (xx >= 0 && xx < n && yy >= 0 && yy < m && maze[ff][xx][yy] !='*')
        return true;
    return false;
}

bool bfs(){
    queue<node> q;
    node s = {sf, sx, sy, 0};
    q.push(s);
    while(!q.empty()){
        node nn = q.front();
        q.pop();
        if (nn.t > T) continue;
        for (int i = 0; i < 4; i++){
            int nx = nn.x + dir[i][0];
            int ny = nn.y + dir[i][1];
            if (ok(nn.f, nx, ny) && !vis[nn.f][nx][ny])
                if (maze[nn.f][nx][ny] == '.'){
                    node ne = {nn.f, nx, ny, nn.t+1};
                    q.push(ne);
                    vis[nn.f][nx][ny] = 1;
                }
                else if (maze[nn.f][nx][ny] == '#' && maze[!nn.f][nx][ny] != '*' && maze[!nn.f][nx][ny] != '#'){
                    if (maze[!nn.f][nx][ny] == 'P' && nn.t + 1 <= T) return true;
                    node ne = {!nn.f, nx, ny, nn.t+1};
                    q.push(ne);
                    vis[nn.f][nx][ny] = 1;
                    vis[!nn.f][nx][ny] = 1;
                }
                else if (maze[nn.f][nx][ny] == 'P' && nn.t + 1 <= T)
                   return true;
        }
    }
    return false;
}

int main()
{
    int c;
    cin>>c;
    while(c--){
        M(maze, 0);
        M(vis, 0);
        cin>>n>>m>>T;
        for (int k = 0; k < 2; k++)
            for (int i = 0; i < n; i++)
                for (int j = 0; j < m; j++){
                    cin>>maze[k][i][j];
                    if (maze[k][i][j] == 'S'){
                        sf = k;
                        sx = i;
                        sy = j;
                    }
                }
        bool flag = bfs();
        if (flag) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }

    return 0;
}