胜利大逃亡 --- bfs记录

Problem Description
Ignatius被魔王抓走了,有一天魔王出差去了,这可是Ignatius逃亡的好机会.
魔王住在一个城堡里,城堡是一个ABC的立方体,可以被表示成A个B*C的矩阵,刚开始Ignatius被关在(0,0,0)的位置,离开城堡的门在(A-1,B-1,C-1)的位置,现在知道魔王将在T分钟后回到城堡,Ignatius每分钟能从一个坐标走到相邻的六个坐标中的其中一个.现在给你城堡的地图,请你计算出Ignatius能否在魔王回来前离开城堡(只要走到出口就算离开城堡,如果走到出口的时候魔王刚好回来也算逃亡成功),如果可以请输出需要多少分钟才能离开,如果不能则输出-1.
Input
输入数据的第一行是一个正整数K,表明测试数据的数量.每组测试数据的第一行是四个正整数A,B,C和T(1<=A,B,C<=50,1<=T<=1000),它们分别代表城堡的大小和魔王回来的时间.然后是A块输入数据(先是第0块,然后是第1块,第2块…),每块输入数据有B行,每行有C个正整数,代表迷宫的布局,其中0代表路,1代表墙.(如果对输入描述不清楚,可以参考Sample Input中的迷宫描述,它表示的就是上图中的迷宫)

特别注意:本题的测试数据非常大,请使用scanf输入,我不能保证使用cin能不超时.在本OJ上请使用Visual C++提交.

Output
对于每组测试数据,如果Ignatius能够在魔王回来前离开城堡,那么请输出他最少需要多少分钟,否则输出-1.

Sample Input
1
3 3 4 20
0 1 1 1
0 0 1 1
0 1 1 1
1 1 1 1
1 0 0 1
0 1 1 1
0 0 0 0
0 1 1 0
0 1 1 0

Sample Output
11

起始状态: 起点(0, 0, 0)
目标状态: 终点(A-1, B-1, C-1)
移动规则: r[6][3] = {{0, -1, 0}, {0, 1, 0}, {1, 0, 0}, {-1, 0, 0}, {0, 0, 1}, {0, 0, -1}};
利用规则(移动方式）生成下一层的所有状态节点（像树），并依次放入队列中，层层往下展开，直到到达终点

#include<bits/stdc++.h>
using namespace std;

int K, T, A, B, C;
int r[6][3] = {{0, -1, 0}, {0, 1, 0}, {1, 0, 0}, {-1, 0, 0}, {0, 0, 1}, {0, 0, -1}};
// vis 标记坐标是否到达过， mp存放地图
int vis[55][55][55], mp[55][55][55];

struct node{
	int x, y, z, step;
};

bool check(int x, int y, int z) {  // 判断是否合法
	if(vis[x][y][z])	return false;
	if(x < 0 || x >= A)	return false;
	if(y < 0 || y >= B)	return false;
	if(z < 0 || z >= C)	return false;
	if(mp[x][y][z])	return false;
	return true; 
}

void bfs() {
	node cur, next;
	queue<node> q;
	cur.x = cur.y = cur.z = cur.step = 0;  // 初始化起点
	vis[0][0][0] = 1;  // 标记
	q.push(cur);
	while(!q.empty()) {
		cur = q.front();
		q.pop();
		if((cur.x == A - 1) && (cur.y == B - 1) && (cur.z == C - 1)) {
			if(cur.step <= T) {  // 注意时间必须小于魔王回来的时间
				printf("%d\n", cur.step);
				return ;
			}
		}
		for(int i = 0; i < 6; i++) {
			next.x = cur.x + r[i][0];
			next.y = cur.y + r[i][1];
			next.z = cur.z + r[i][2];
			if(!check(next.x, next.y, next.z))	continue;
			next.step = cur.step + 1;
			vis[next.x][next.y][next.z] = 1;
			q.push(next);
		}
	}
	// while中遍历所有状态，若未在while中退出，则说明不可到达
	printf("-1\n");
	return ;
}

int main() {
	scanf("%d", &K);
	while(K--) {
		memset(vis, 0, sizeof(vis));
		scanf("%d %d %d %d", &A, &B, &C, &T);
		int a, b, c, d; 
		for(int i = 0; i < A; i++) {
			for(int j = 0; j < B; j++) {
				for(int k = 0; k < C;k++) {
					scanf("%d", &mp[i][j][k]);
					//printf("mp[%d][%d][%d] = %d\n", i, j, k, mp[i][j][k]);
				}
			}
		}
		bfs();
	}
	return 0;
}