王道机试练习——胜利大逃亡之广度优先搜索

王道机试练习——胜利大逃亡之广度优先搜索

题目描述

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

代码

#include<stdio.h>
#include<queue>
using namespace std;
bool mark[50][50][50];
int maze[50][50][50];//保存立方体信息
struct N {//
	int x, y, z;
	int t;
};
queue<N> Q;//队列中的元素为状态
int go[][3] = {//坐标变换数组，由坐标扩展得到的新坐标均可通过(x+go[i][0],y+go[i][1],z+go[i][2])
	1,0,0,
	-1,0,0,
	0,1,0,
	0,-1,0,
	0,0,1,
	0,0,-1
};
int BFS(int a, int b, int c) {//广度优先搜索
	while (Q.empty() == false) {//当队列中仍有元素可以扩展时循环
		N now = Q.front();//得到队头状态
		Q.pop();//从队列弹出队头状态
		for (int i = 0; i < 6; i++) {//依次扩展其六个相邻节点
			int nx = now.x + go[i][0];
			int ny = now.y + go[i][1];
			int nz = now.z + go[i][2];
			if (nx < 0 || nx >= a || ny < 0 || ny >= b || nz < 0 || nz >= c) continue;
			if (maze[nx][ny][nz] == 1) continue;//该位置为墙，则丢弃
			if (mark[nx][ny][nz] == 1) continue;//包含该位置的状态已经被得到，丢弃
			N tmp;//新状态
			tmp.x = nx;
			tmp.y = ny;
			tmp.z = nz;//新状态包含的坐标
			tmp.t = now.t + 1;//新状态的耗时
			Q.push(tmp);//将该状态放入队列
			mark[nx][ny][nz] = true;//标记该坐标
			if (nx == a - 1 && ny == b - 1 && nz == c - 1) return tmp.t;//若改坐标为终点，直接返回其耗时
		}
	}
	return -1;//所有状态被查找完后，仍得不到所需坐标，则返回-1
}
int main() {
	int T;
	scanf("%d", &T);
	while (T--) {
		int a, b, c, t;
		scanf("%d%d%d%d", &a, &b, &c, &t);//输入
		for (int i = 0; i < a; i++) {
			for (int j = 0; j < b; j++) {
				for (int k = 0; k < c; k++) {
					scanf("%d", &maze[i][j][k]);//输入立方体信息
					mark[i][j][k] = false;//初始化数组信息
				}
			}
		}
		while (Q.empty() == false) Q.pop();
		mark[0][0][0] = true;
		N tmp;
		tmp.t = tmp.y = tmp.x = tmp.z = 0;
		Q.push(tmp);
		int rec = BFS(a, b, c);
		if (rec <= t) printf("%d\n", rec);
		else printf("-1\n");
	}
	return 0;
}