hdu2102 BFS

Posted 2020-07-02 曹孟德

        这是一道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[2][MAX_SIZE][MAX_SIZE];
char map[2][MAX_SIZE][MAX_SIZE];
int dir[4][2] = { {0,1},{0,-1},{1,0},{-1,0} };
struct Point{
    int x, y, z, step;
    bool Isleg() {
        if ( y<0 || y>N-1||z<0||z>M-1||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 = 0; i <N; i++)
            scanf("%s", map[0][i]);
        for (i =0; i < N;i++)
            scanf("%s",map[1][i]);
        res = BFS();
        if (res <= T)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}
int BFS() {
    Point next, pos;
    int k,i,j;
    for (i = 0; i < 2; i++) {
        for (k = 0; k <N; k++)
            for (j = 0; j <M; j++)
                visit[i][k][j] = 0;
    }
    pos.x = 0, pos.y = 0,pos.z=0;     //入口位置
    pos.step = 0;
    queue<Point>Q;
    Q.push(pos);                      //入口位置首先入队
    visit[pos.x][pos.y][pos.z] = 1;   //入口位置标记为访问
    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 + 1) % 2;       //进入下一层
            if (next.Isleg()) {
                visit[next.x][next.y][next.z] = 1;
                next.step--;         //时空传输不发费时间
                Q.push(next);
            }
        }
        else {
            for (k = 0; k < 4; k++) {     //扫描四个方向
                next = pos;
                next.y += dir[k][0];
                next.z += dir[k][1];
                if (next.Isleg()) {
                    visit[next.x][next.y][next.z] = 1;
                    Q.push(next);
                }
            }
        }
    }
    return INT_MAX;
}