DFS,回溯,用stack存储路径
问题描述
给出一个n*m的迷宫,起点为(0,0),终点为(n-1,m-1).可以向上、下、左、右四个方向走.
任务:判断是否有可行路径。如果有可行路径,输出所有的可行路径.
(0代表通道,1代表墙壁)
Sample
5 6
0 1 0 0 0 1
0 0 0 1 0 0
1 0 1 0 0 1
1 0 0 1 0 1
1 1 0 0 0 0
code
#include<iostream>
#include<stack>
using namespace std;
int maze[10][10];//迷宫
int vis[10][10];//记录迷宫中的某个位置是否访问过
int n,m;
int dir[4][2] = {{0,1},{0,-1},{1,0},{-1,0}};//四个方向
struct point//位置
{
int x,y;
} p;
stack<point> path,temp;//记录路径,temp是一个临时变量,和path一起处理路径
int count;//路径条数
void dfs(int x,int y)//x,y:当前位置
{
if(x==n-1 && y==m-1)//成功---下面处理路径问题
{
cout << "******************路径"<< ++count << "******************" << endl;
while(!path.empty())//将path里面的点取出来,放在temp里面
{//path从栈顶-栈底的方向,路径是从终点-起点的顺序
point p1 = path.top();
path.pop();
temp.push(p1);
}
while(!temp.empty())
{//输出temp里面的路径,这样刚好是从起点到终点的顺序
point p1 = temp.top();
temp.pop();
path.push(p1);//将路径放回path里面,因为后面还要回溯!!!
cout << "(" << p1.x << "," << p1.y << ")" << endl;
}
return;
}
if(x<0 || x>=n || y<0 || y>=m)//越界
return;
//如果到了这一步,说明还没有成功,没有出界
for(int i=0;i<4;i++)//从4个方向探测
{
int nx = x + dir[i][0];
int ny = y + dir[i][1];//nx,ny:选择一个方向,前进一步之后,新的坐标
//条件:nx,ny没有出界,maze[nx][ny]=0这个点不是障碍可以走,vis[nx][ny]=0说明(nx,ny)没有访问过,可以访问
if(0<=nx && nx<n && 0<=ny && ny<m && maze[nx][ny]==0 && vis[nx][ny]==0)
{
vis[nx][ny]=1;//设为访问过
p.x = nx;
p.y = ny;
path.push(p);//让当前点进栈
dfs(nx,ny);//进一步探测
vis[nx][ny]=0;//回溯
path.pop();//由于是回溯,所以当前点属于退回去的点,需要出栈
}
}
}
int main()
{
count = 0;
p.x = 0;
p.y = 0;
path.push(p);//起点先入栈
cin >> n >> m;
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
vis[i][j] = 0;
cin >> maze[i][j];
}
}
vis[0][0]=1;
//起点的vis要在这里为1,因为在dfs中不处理此点
dfs(0,0);
return 0;
}
vis数组的变化:在if分支里添加输出语句
if(0<=nx && nx<n && 0<=ny && ny<m && maze[nx][ny]==0 && vis[nx][ny]==0)
{
vis[nx][ny]=1;//设为访问过
p.x = nx;
p.y = ny;
path.push(p);//让当前点进栈
cout<<endl;
for(int o=0;o<n;o++){
for(int u=0;u<m;u++){
cout<<vis[o][u]<<' ';
}
cout<<endl;
}
dfs(nx,ny);//进一步探测
vis[nx][ny]=0;//回溯
path.pop();//由于是回溯,所以当前点属于退回去的点,需要出栈
}
5 6
0 1 0 0 0 1
0 0 0 1 0 0
1 0 1 0 0 1
1 0 0 1 0 1
1 1 0 0 0 0
1 0 0 0 0 0
1 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 0 0 0 0
1 1 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 0 0 0 0
1 1 1 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 1 0 0 0
1 1 1 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 1 1 0 0
1 1 1 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 1 1 1 0
1 1 1 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 1 1 1 0
1 1 1 0 1 1
0 0 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 1 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 1 1 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 1 0
0 0 0 0 1 0
0 0 0 0 0 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 1 0
0 0 0 0 1 0
0 0 0 0 1 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 1 0
0 0 0 0 1 0
0 0 0 0 1 1
******************路径1******************
(0,0)
(1,0)
(1,1)
(1,2)
(0,2)
(0,3)
(0,4)
(1,4)
(2,4)
(3,4)
(4,4)
(4,5)
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 1 0
0 0 0 0 1 0
0 0 0 1 1 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 1 0
0 0 0 0 1 0
0 0 1 1 1 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 1 0
0 0 1 0 1 0
0 0 1 1 1 0
1 0 1 1 1 0
1 1 1 0 1 0
0 0 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 1 1 1 0
1 1 1 0 1 0
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 1 0 0 0 0
0 0 0 0 0 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 1 1 0 0 0
0 0 0 0 0 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 1 1 0 0 0
0 0 1 0 0 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 1 1 0 0 0
0 0 1 1 0 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 1 1 0 0 0
0 0 1 1 1 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 1 1 0 0 0
0 0 1 1 1 1
******************路径2******************
(0,0)
(1,0)
(1,1)
(2,1)
(3,1)
(3,2)
(4,2)
(4,3)
(4,4)
(4,5)
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 0 0 0 0
1 1 0 0 0 0
0 1 0 1 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 0 0 0 0
1 1 0 0 1 0
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 0 0 0 0
1 1 0 0 1 1
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 0 0 1 0
1 1 0 0 1 0
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 0 1 1 0
1 1 0 0 1 0
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 1 1 1 0
1 1 0 0 1 0
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
1 0 1 1 1 0
1 1 1 0 1 0
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 0
-
由于方向向量的存在,for循环一次后换一个方向,所以在vis=0回溯会不会走原来的路。
-
回溯就把结点从栈中弹出。
如:
1 0 1 1 1 0
1 1 1 0 1 0
0 1 0 0 1 0
0 1 1 0 1 0
0 0 1 1 1 01 0 0 0 0 0
1 1 0 0 0 0
0 1 0 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0