Problem
定義一個(gè)二維數(shù)組:
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,
};
它表示一個(gè)迷宮么介,其中的1表示墻壁,0表示可以走的路,只能橫著走或豎著走囚灼,不能斜著走,要求編程序找出從左上角到右下角的最短路線祭衩。
Input
一個(gè)5 × 5的二維數(shù)組灶体,表示一個(gè)迷宮。數(shù)據(jù)保證有唯一解掐暮。
Output
左上角到右下角的最短路徑蝎抽,格式如樣例所示。
Sample Input
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
Sample Output
(0, 0)
(1, 0)
(2, 0)
(2, 1)
(2, 2)
(2, 3)
(2, 4)
(3, 4)
(4, 4)
思路
迷宮最短路模板題,借助queue樟结,BFS即可养交,用stack輔助記錄路徑。
代碼
#include<iostream>
#include<queue>
#include<stack>
using namespace std;
int maze[5][5],vis[5][5]={0},dist[5][5];
int dx[4]={-1,1,0,0};
int dy[4]={0,0,-1,1};
struct Node {
int x,y;
Node(int x,int y):x(x),y(y){}
Node(){}
}pre[5][5];
queue<Node> Q;
void BFS() {
while(!Q.empty()) Q.pop();
dist[0][0]=0;
vis[0][0]=1;
Q.push(Node(0,0));
while(!Q.empty()) {
Node node=Q.front();Q.pop();
int x=node.x,y=node.y;
for(int d=0;d<4;d++) {
int nx=x+dx[d];
int ny=y+dy[d];
if(nx>=0&&nx<5&&ny>=0&&ny<5&&vis[nx][ny]==0&&maze[nx][ny]==0) {
vis[nx][ny]=1;
Q.push(Node(nx,ny));
dist[nx][ny]=1+dist[x][y];
pre[nx][ny]=Node(x,y);
if(nx==4&&ny==4) return;
}
}
}
}
int main() {
for(int i=0;i<5;i++)
for(int j=0;j<5;j++)
cin>>maze[i][j];
BFS();
stack<Node> S;
int cur_x=4,cur_y=4;
while(1){
S.push(Node(cur_x,cur_y));
if(cur_x==0&&cur_y==0) break;
int x=cur_x,y=cur_y;
cur_x=pre[x][y].x;
cur_y=pre[x][y].y;
}
while(!S.empty()){
Node node=S.top();
cout<<'('<<node.x<<", "<<node.y<<')'<<endl;
S.pop();
}
return 0;
}