这道题使用BFS做的,刚开始有点不太理解为什么使用队列,一旦遇到可以到达终点的节点就立即返回,找到最短时间,最后明白了,因为在队列里的所有节点一定比队头节点
的时间长。下面是具体代码:
#include<stdio.h>
#include<queue>
using namespace std;
typedef struct{ int x,y,z,steps;
}point;
point start,end;
int a,b,c,t,n;
int map[51][51][51];
int dir[6][3]={{1,0,0}, {-1,0,0}, {0,1,0}, {0,-1,0}, {0,0,1}, {0,0,-1}};
int bfs(point start){ queue<point>q; int i; point cur,next; if(start.x==a-1&&start.y==b-1&&start.z==c-1)//考虑起点和终点相同的情况 { return 0; } start.steps=0; map[start.x][start.y][start.z]=1; q.push(start); while(!q.empty()) { cur=q.front();//取队首元素 q.pop(); for(i=0;i<6;i++) //广度优先搜索 { next.x=cur.x+dir[i][0]; next.y=cur.y+dir[i][1]; next.z=cur.z+dir[i][2]; if(next.x==a-1 && next.y==b-1 && next.z==c-1) //下一步就是目的地 { return cur.steps+1; }if(next.x>=0&&next.x<a&&next.y>=0&&next.y<b&&next.z>=0&&next.z<c) if(map[next.x][next.y][next.z]!=1) { map[next.x][next.y][next.z]=1; next.steps=cur.steps+1; q.push(next); } } } return -1;
}
int main()
{ int i,j,k,step; scanf("%d\n",&n); while(n--) { scanf("%d %d %d %d",&a,&b,&c,&t); for(i=0;i<a;i++) for(j=0;j<b;j++) for(k=0;k<c;k++) scanf("%d",&map[i][j][k]); if(a+b+c-3>t) {printf("-1\n");continue; } if(map[a-1][b-1][c-1]==1){printf("-1\n");continue;}start.x=0; start.y=0; start.z=0;step=bfs(start); if(step>=0&&step<=t) printf("%d\n",step); else printf("-1\n"); } return 0;
}