hdu2102 BFS

这是一道BFS的搜索题目，只是搜索范围变为了三维。定义数组visit[x][y][z]来标记空间位置，x表示楼层，y和z表示相应楼层的平面坐标。

#define _CRT_SECURE_NO_DEPRECATE
#include<iostream>
using namespace std;
#include<queue>
#define MAX_SIZE 12
int N, M, T;
char visit[][MAX_SIZE][MAX_SIZE];
char map[][MAX_SIZE][MAX_SIZE];
int dir[][] = { {,},{,-},{,},{-,} };
struct Point{
    int x, y, z, step;
    bool Isleg() {
        if ( y< || y>N-||z<||z>M-||map[x][y][z]=='*'||visit[x][y][z])
            return false;
        return true;
    }
};
int BFS();
int main() {
    int i,c,res;
    scanf("%d", &c);
    while(c--){
        scanf("%d%d%d", &N, &M, &T);
        for (i = ; i <N; i++)
            scanf("%s", map[][i]);
        for (i =; i < N;i++)
            scanf("%s",map[][i]);
        res = BFS();
        if (res <= T)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return ;
}
int BFS() {
    Point next, pos;
    int k,i,j;
    for (i = ; i < ; i++) {
        for (k = ; k <N; k++)
            for (j = ; j <M; j++)
                visit[i][k][j] = ;
    }
    pos.x = , pos.y = ,pos.z=;     //入口位置
    pos.step = ;
    queue<Point>Q;
    Q.push(pos);                      //入口位置首先入队
    visit[pos.x][pos.y][pos.z] = ;   //入口位置标记为访问
    while (!Q.empty()) {
        pos = Q.front();
        Q.pop();
        if (map[pos.x][pos.y][pos.z] == 'P')
            return  pos.step;
        pos.step++;
        if (map[pos.x][pos.y][pos.z] == '#') {
            next = pos;
            next.x = (next.x + ) % ;       //进入下一层
            if (next.Isleg()) {
                visit[next.x][next.y][next.z] = ;
                next.step--;         //时空传输不发费时间
                Q.push(next);
            }
        }
        else {
            for (k = ; k < ; k++) {     //扫描四个方向
                next = pos;
                next.y += dir[k][];
                next.z += dir[k][];
                if (next.Isleg()) {
                    visit[next.x][next.y][next.z] = ;
                    Q.push(next);
                }
            }
        }
    }
    return INT_MAX;
}