用栈来求解迷宫问题,和用递归来求解迷宫问题,其实如出一辙,底层都是用到回溯。
思路:
拿到当前迷宫盒子时,我们有四个选择去走,我们就试一试所有路,如果说有一条路,到尽头的话,就称这为死路,我们得做一个标记,在迷宫上,这样子的话,如果回溯到上一级,就不会再走这条路了。
这里先设一个类表示迷宫格子
i 表示 行
j 表示列
di表示方位
class Box {
int i;
int j;
int di;
public Box(int i1,int j1,int di1) {
i = i1;
j = j1;
di = di1;
}
}
这里我们设往上走,di = 0
往右走 di = 1
往下走 di = 2
往左走 di = 3
设置一下迷宫,方法
public void Setmg(int[][] a) {
for(int i = 0;i < m;i++) {
for(int j = 0; j < n; j++) {
mg[i][j] = a[i][j];
}
}
}
写一个方法,打印迷宫
public void printMg() {
for(int i = 0; i < m; i++) {
for(int j = 0; j < n; j++) {
System.out.print((mg[i][j]) + " ");
}
System.out.println();
}
}
核心代码
boolean mgpath(int xi,int yi,int xe,int ye) {
int i,j,di,i1 = 0,j1 = 0;
//建立一个空栈
Stack<Box> st = new Stack<Box>();
//辅助变量
boolean find;
Box box,e;
//将入口进栈
st.push(new Box(xi,yi,-1));
//将进栈的值设为-1
mg[xi][yi] = -1;
while(!st.empty()) {
//取得栈顶的元素
box = st.peek();
i = box.i;
j = box.j;
di = box.di;
//判断是否为出口
if(i == xe && j == ye) {
//将出口设为2
mg[i][j] = 2;
while(!st.empty()) {
e = st.pop();
mg[e.i][e.j] = 2;
}
return true;
}
find = false;
//循环找到下一个出口
while(di < 3 && !find) {
di++;
switch (di) {
case 0: i1 = i -1; j1 = j; break;
case 1: i1 = i; j1 = j + 1; break;
case 2: i1 = i + 1; j1 = j; break;
case 3: i1 = i; j1 = j - 1; break;
}
if(mg[i1][j1] == 0) {
find = true;
}
}
//find为true的话,说明找到了下一个,将栈顶的di元素改为此处的di,保留方位
if(find) {
e = st.pop();
e.di = di;
st.push(e);
st.push(new Box(i1,j1,-1));
mg[i1][j1] = -1;
} else {//find为false说明此节点为死路
mg[i][j] = 0 ;
st.pop();
}
}
return false;
}