01迷宫及路径记录(DFS&&BFS)
题意
给出一个N*M的迷宫图和一个入口、一个出口。
编一个程序,打印一条从迷宫入口到出口的路径。这里黑色方块的单元表示走不通(用-1表示),白色方块的单元表示可以走(用0表示)。只能往上、下、左、右四个方向走。输出一条可行的路径,如果无路则输出“no way.”。
分析
本题为搜索模版题,bfs和dfs均可,主要考虑如何存储路径。
- 对于dfs,代码里使用way[stp][1],way[stp][2]分别表示第stp步走到的点坐标x,y,走到终点输出即可。
- 对于bfs,队列的结构体中多开一个变量stp,表示走到每个点走的步数,每个点的stp由上一个点的stp更新来。
由于只输出一条路径,当输出一次之后用flag标记,不再进行之后的搜索。
代码
#include<bits/stdc++.h> using namespace std; const int N = 1005,M = 1005; int n,m; int mp[N][M]; int way[M+N][3],cnt; int ix,iy,ox,oy; struct pos{ int x,y,stp; }; queue<pos>q; int dx[4]={1,-1,0,0}; int dy[4]={0,0,1,-1}; bool vis[N][M],flag; inline void print(int stp) { for(int i=1;i<stp;i++) printf("%d,%d\n",way[i][1],way[i][2]); puts(""); } inline void bfs() { q.push((pos){ix,iy,2}); way[1][1]=ix,way[1][2]=iy; while(!q.empty()) { pos now=q.front(); int stp=now.stp; q.pop(); if(now.x==ox&&now.y==oy) { way[stp][1]=now.x; way[stp][2]=now.y; print(stp); flag=1; return; } for(int i=0;i<4;i++) { int nx=now.x+dx[i],ny=now.y+dy[i]; // if(flag) return; if(nx<=n&&ny<=m&&nx>=1&&ny>=1&&mp[nx][ny]==0&&!vis[nx][ny]) { vis[nx][ny]=1; way[stp][1]=nx,way[stp][2]=ny; q.push((pos){nx,ny,stp+1}); } } } } void dfs(int x,int y,int stp) { if(flag) return; if(x==ox&&y==oy) { way[stp][1]=x; way[stp][2]=y; print(stp); flag=1; return; } for(int i=0;i<4;i++) { int nx=x+dx[i],ny=y+dy[i]; if(flag) return; if(nx<=n&&ny<=m&&nx>=1&&ny>=1&&mp[nx][ny]==0&&!vis[nx][ny]) { vis[nx][ny]=1; way[stp][1]=nx,way[stp][2]=ny; dfs(nx,ny,stp+1); vis[nx][ny]=0; } } } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) scanf("%d",&mp[i][j]); scanf("%d%d%d%d",&ix,&iy,&ox,&oy); way[1][1]=ix,way[1][2]=iy; dfs(ix,iy,2); // bfs(); if(!flag) printf("no way."); return 0; } /* 8 5 -1 -1 -1 -1 -1 0 0 0 0 -1 -1 -1 -1 0 -1 -1 0 0 0 -1 -1 0 0 -1 -1 -1 0 0 0 -1 -1 -1 -1 0 -1 -1 0 0 0 -1 2 1 8 4 */
如果要记录所有路径,把dfs中关于if(flag)return; 的两行删去即可,bfs可能比较复杂(我不会)