POJ 2225 && HDU 1240 Asteroids!(bfs)

Description
给定一个三维的迷宫,已知起点和终点,问是否能从起点到达终点,并且最短的距离又是多少
Input
第一行为START n,n为三维迷宫的边长,之后为n个矩阵,每个矩阵为一层迷宫,矩阵顺序按层数从下到上,之后两行每行三个整数分别表示起点和终点的横纵竖坐标,然后以END结束一组用例的输入,以文件尾结束全部输入
Output
对于每组用例,若能从起点到达终点则输出最短距离,否则输出NO ROUTE
Sample Input
START 1
o
0 0 0
0 0 0
END
START 3
XXX
XXX
XXX
ooo
ooo
ooo
XXX
XXX
XXX
0 0 1
2 2 1
END
START 5
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
XXXXX
XXXXX
XXXXX
XXXXX
XXXXX
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
ooooo
0 0 0
4 4 4
END
Sample Output
1 0
3 4
NO ROUTE
Solution
三维迷宫,简单dfs
Code

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
struct data
{
    int x,y,z;
    bool operator==(const data &ne)const
    {
        return x==ne.x&&y==ne.y&&z==ne.z;
    }
    data(){}
    data(int _x,int _y,int _z){x=_x;y=_y;z=_z;}
};
int mark[15][15][15];
char map[15][15][15];//第一维为竖坐标,第二维为纵坐标,第三维为横坐标 
int dr[][3]={0,0,1,0,0,-1,0,1,0,0,-1,0,1,0,0,-1,0,0};//上下左右前后六个方向 
int main()
{
    char s[10];
    while(scanf("%s",s)!=EOF)
    {
        int n;
        scanf("%d",&n);
        for(int i=0;i<n;i++)//输入迷宫 
            for(int j=0;j<n;j++)
                scanf("%s",map[i][j]);
        data a,t;
        scanf("%d%d%d",&a.z,&a.y,&a.x);//起点坐标,注意顺序 
        scanf("%d%d%d",&t.z,&t.y,&t.x);//终点坐标,注意顺序 
        queue<data> Q;
        memset(mark,-1,sizeof(mark));//初始化 
        mark[a.x][a.y][a.z]=0;//记录步数 
        Q.push(a);//将起点压入队列 
        scanf("%s",s);//输入END 
        bool flag=false;//标志变量 
        while(!Q.empty())
        {
            a=Q.front();//出列 
            Q.pop(); 
            if(a==t)//到达终点 
            {
                flag=true;
                break;
            }
            int tp=mark[a.x][a.y][a.z];
            for(int i=0;i<6;i++)//六个方向枚举 
            {
                data b(a.x+dr[i][0],a.y+dr[i][1],a.z+dr[i][2]);
                if(b.x<0||b.x>=n||b.y<0||b.y>=n||b.z<0||b.z>=n||mark[b.x][b.y][b.z]!=-1||map[b.x][b.y][b.z]=='X')
                    continue;
                mark[b.x][b.y][b.z]=tp+1;//步数加一 
                Q.push(b);//压入队列 
            }
        }
        if(flag)//能够到达终点 
            printf("%d %d\n",n,mark[t.x][t.y][t.z]);
        else//不能到达终点 
            printf("NO ROUTE\n");
    }
    return 0;
}

你可能感兴趣的:(POJ 2225 && HDU 1240 Asteroids!(bfs))