HDU 1253 三维数组的图上找最短路

题目大意:

从三维空间的(0,0,0)出发到（a-1,b-1,c-1)，每移动一个都要时间加一，计算最短时间

根据六个方向，开个bfs，像spfa那样计算最短路径就行了，但是要1200多ms，也不知道有没有更好的方法

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <iostream>
 4 #include <queue>
 5 using namespace std;
 6 int dp[52][52][52] , a , b , c;
 7 bool wall[52][52][52] , vis[52][52][52];
 8 int dir[6][3] = {{0,0,1} , {0,1,0} , {1,0,0} , {0,0,-1} , {0,-1,0} , {-1,0,0}};
 9 struct Node{
10     int x , y , z;
11     Node(int x = 0 , int y = 0 , int z = 0):x(x),y(y),z(z){}
12 };
13 queue<Node> q;
14
15 bool ok(int x , int y , int z)
16 {
17     if(x<0 || x>=a) return false;
18     if(y<0 || y>=b) return false;
19     if(z<0 || z>=c) return false;
20     if(wall[x][y][z]) return false;
21     return true;
22 }
23
24 int main()
25 {
26   //  freopen("a.in" , "r" , stdin);
27     int T , t , w;
28     scanf("%d" , &T);
29     while(T--){
30         scanf("%d%d%d%d" , &a , &b , &c , &t);
31         for(int i = 0 ; i<a ; i++)
32             for(int j = 0 ; j<b ; j++)
33                 for(int k = 0 ; k<c ; k++){
34                     scanf("%d" , &w);
35                     if(w) wall[i][j][k] = true;
36                     else wall[i][j][k] = false;
37                 }
38         memset(dp , 0x3f , sizeof(dp));
39         memset(vis , 0 , sizeof(vis));
40         dp[0][0][0] = 0;
41         q.push(Node(0,0,0));
42         while(!q.empty()){
43             Node u = q.front();
44             q.pop();
45             vis[u.x][u.y][u.z] = false;
46             for(int i = 0 ; i<6 ; i++){
47                 Node v;
48                 v.x = u.x+dir[i][0];
49                 v.y = u.y+dir[i][1];
50                 v.z = u.z+dir[i][2];
51                 if(ok(v.x,v.y,v.z)){
52                     if(dp[v.x][v.y][v.z] > dp[u.x][u.y][u.z] + 1){
53                         dp[v.x][v.y][v.z] = dp[u.x][u.y][u.z] + 1;
54                         if(!vis[v.x][v.y][v.z]){
55                             q.push(v);
56                             vis[v.x][v.y][v.z] = true;
57                         }
58                     }
59                 }
60             }
61         }
62
63         if(dp[a-1][b-1][c-1] > t) puts("-1");
64         else printf("%d\n" , dp[a-1][b-1][c-1]);
65     }
66     return 0;
67 }