题目
假设一个探险家被困在了地底的迷宫之中,要从当前位置开始找到一条通往迷宫出口的路径。迷宫可以用一个二维矩阵组成,有的部分是墙,有的部分是路。迷宫之中有的路上还有门,每扇门都在迷宫的某个地方有与之匹配的钥匙,只有先拿到钥匙才能打开门。请设计一个算法,帮助探险家找到脱困的最短路径。如前所述,迷宫是通过一个二维矩阵表示的,每个元素的值的含义如下 0-墙,1-路,2-探险家的起始位置,3-迷宫的出口,大写字母-门,小写字母-对应大写字母所代表的门的钥匙
输入描述:
迷宫的地图,用二维矩阵表示。第一行是表示矩阵的行数和列数M和N
后面的M行是矩阵的数据,每一行对应与矩阵的一行(中间没有空格)。M和N都不超过100, 门不超过10扇。
输出描述:
路径的长度,是一个整数
示例1
输入
5 5
02111
01a0A
01003
01001
01111
输出
7
思路
常规的BFS寻路,要多考虑一个因素是钥匙。需要知道只有当前位置和当前所拥有的钥匙才能唯一确定是否已经走过(而不能单单靠位置来判断),再者是钥匙的存储方式(当前拥有钥匙可能不只一把)。
采用二进制存储,由题意知最多十把钥匙,十位二进制中每位的 1 代表此种钥匙已拥有。可通过 1 的左移实现。
对于拿钥匙,把 当前钥匙 与 已经拥有的钥匙key 作 或运算;对于开门,把 拥有的钥匙key 与 当前门所对应的数 进行 与运算,若结果为0则表示没有匹配成功。
import java.util.*;
public class Main{
static int n,m;
static char[][] map;
static int[] pos;//01 起始位置
public static void main(String[] args){
Scanner cin = new Scanner(System.in);
String mn = cin.nextLine();
m = Integer.parseInt(mn.split(" ")[0]);
n = Integer.parseInt(mn.split(" ")[1]);
map = new char[m][n];
pos = new int[2];
for(int i=0; i<m; i++){
String t = cin.next();
for(int j=0; j<n; j++){
map[i][j] = t.charAt(j);
if(map[i][j] == '2'){
pos[0] = i;
pos[1] = j;
}
}
}
int sum = BFS();
System.out.println(sum);
}
public static int BFS(){
int[][] dir = {{0,1}, {0,-1}, {1,0},{-1,0}};
boolean[][][] vis = new boolean[m][n][1024]; //用当前位置和所拥有钥匙来唯一确定是否走过
Queue<Dis> que = new LinkedList<Dis>();
vis[pos[0]][pos[1]][0] = true;
que.add(new Dis(pos[0], pos[1], 0, 0));
while( !que.isEmpty()){
Dis d = que.poll();
for(int i=0; i<4; i++){
int u = d.u + dir[i][0];
int v = d.v + dir[i][1];
int key = d.key;
if(FlaseOrTrue(u, v) == false) //越界、不可走
continue;
if(map[u][v] == '3') //到达出口
return d.cnt+1;
if(map[u][v]>='a' && map[u][v]<='z') //遇到钥匙拿起
key = key | (1 << (map[u][v]-'a'));
if(map[u][v]>='A' && map[u][v]<='Z'){ //遇到门
if((key & (1 << (map[u][v]-'A'))) == 0) //没有匹配到对应钥匙
continue;
}
if(vis[u][v][key] == false) {
vis[u][v][key] = true;
que.add(new Dis(u, v, key, d.cnt+1));
}
}
}
return -1;
}
static boolean FlaseOrTrue(int u, int v){
if(u<0 || u>=m || v<0 || v>=n) return false;
if(map[u][v] == '0') return false;
return true;
}
static class Dis{
int u, v, key, cnt;
public Dis(int u, int v, int key, int cnt){
this.u = u;
this.v = v;
this.key = key;
this.cnt = cnt;
}
}
}
如有错误或不合理的地方,敬请指正~
加油!!