bfs poj3984 迷宮最短路徑且輸出路徑
阿新 • • 發佈:2018-11-11
題意:
定義一個二維陣列:
int maze[5][5] = {
0, 1, 0, 0, 0,
0, 1, 0, 1, 0,
0, 0, 0, 0, 0,
0, 1, 1, 1, 0,
0, 0, 0, 1, 0,
};
它表示一個迷宮,其中的1表示牆壁,0表示可以走的路,只能橫著走或豎著走,不能斜著走,要求程式設計序找出從左上角到右下角的最短路線。
分析:
典型的BFS應用,要你求從左上角到右下角的最短路徑,且保證有唯一解,且要輸出路徑.
直接BFS求解即可,需要用到vis陣列和dist陣列,用pre陣列來儲存當前節點的最短路徑上的前一個點(或方向也行).
其實也可以不用pre陣列的,可以直接倒推出最短路徑.
#include<cstdio> #include<cstring> #include<algorithm> #include<queue> #include<stack> using namespace std; int maze[10][10]; int vis[10][10],dist[10][10]; int dr[]={-1,1,0,0};//上,下,左,右 int dc[]={0,0,-1,1}; struct Node { int r,c;//也可以在Node中加一個int pre屬性,然後做一個全域性的nodes,就不用pre[][]陣列了. Node(int r,int c):r(r),c(c){} Node(){} }pre[10][10]; queue<Node> Q; void BFS() { while(!Q.empty()) Q.pop(); memset(vis,0,sizeof(vis)); dist[0][0]=0; vis[0][0]=1; Q.push(Node(0,0)); while(!Q.empty()) { Node node=Q.front();Q.pop(); int r=node.r,c=node.c; for(int d=0;d<4;d++) { int nr=r+dr[d]; int nc=c+dc[d]; if(nr>=0&&nr<5&&nc>=0&&nc<5&&vis[nr][nc]==0&&maze[nr][nc]==0) { vis[nr][nc]=1; Q.push(Node(nr,nc)); dist[nr][nc]=1+dist[r][c]; pre[nr][nc]=Node(r,c); if(nr==4&&nc==4) return ; } } } } int main() { for(int i=0;i<5;i++) for(int j=0;j<5;j++) scanf("%d",&maze[i][j]); BFS(); stack<Node> S; int cur_r=4,cur_c=4; while(true) //這裡也可以重開一個數組記錄。 { S.push(Node(cur_r,cur_c)); if(cur_r==0&&cur_c==0) break; int r=cur_r,c=cur_c; cur_r=pre[r][c].r; cur_c=pre[r][c].c; } while(!S.empty()) { Node node=S.top(); S.pop(); printf("(%d, %d)\n",node.r,node.c); } return 0; }