题意:
给出了机器人要行走的迷宫,还有机器人一次最多能穿过的墙的数目K,求最短的路径。
思路:一开始只是简单的用二维数组标记走过的路径,wa了,后来才看了别人的题解才知道,二维的标记有时会陷入死胡同。
#include<iostream>
#include<cstring>
using namespace std;
int maze[22][22],vis[22][22][22];
struct{int x,y,c,pre;
}a[100000];
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
int m,n,k;
int co,flag;
void pri(int p){if(a[p].pre!=-1){co++;pri(a[p].pre);}
}void bfs(){memset(vis,0,sizeof(vis));int now=1,bu=0;a[0].x=0;a[0].y=0;a[0].pre=-1;a[0].c=0;while(bu<now){for(int i=0;i<4;i++){int sx=a[bu].x+dx[i];int sy=a[bu].y+dy[i];if(sx<0||sy<0||sy>=m||sx>=n) {continue;}if(maze[sy][sx]==1) a[now].c=a[bu].c+1;else a[now].c=0; if(vis[sy][sx][a[now].c]||a[bu].c>k) {continue;}else{vis[sy][sx][a[now].c]=1;a[now].x=sx;a[now].y=sy;a[now].pre=bu;now++;}if(sy==m-1&&sx==n-1){pri(bu); flag=1;return;}}bu++;}
}
int main(){int t;cin>>t; while(t--){co=1;flag=0;cin>>m>>n>>k;for(int i=0;i<m;i++)for(int j=0;j<n;j++)cin>>maze[i][j];bfs();if(m==1&&n==1) cout<<0<<endl; else if(flag) cout<<co<<endl;else cout<<-1<<endl;}
}
我是用数组进行标记,比较繁琐,有大神是通过队列标记的一边入队列一边弹出来,感觉更加简练,附上大神代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
int n,m,k,t;
int map[25][25];
bool vis[25][25][25];//增加一个维度Z来表示跃过的墙的层数。
int dx[4]= {1,0,-1,0};
int dy[4]= {0,1,0,-1};
struct Point
{ int x,y; int step; int layer; Point(int x=1,int y=1,int step=0,int layer=0):x(x),y(y),step(step),layer(layer) {}
};
int ans()
{ queue<Point>Q; memset(vis,0,sizeof(vis)); Point a(1,1,0,0); Point end_point(n,m); Q.push(a); vis[1][1][0]=true; while(!Q.empty()) { Point now=Q.front(); Q.pop(); if(now.x==end_point.x&&now.y==end_point.y) return now.step; for(int i=0; i<4; i++) { int x=dx[i]+now.x; int y=dy[i]+now.y; int layer=now.layer; if(map[x][y]) layer++;//因为每次只会移动一步。而且遇见一层墙,所以加加。 else layer=0; if(layer<=k&&!vis[x][y][layer]&&x>=1&&y>=1&&x<=n&&y<=m) { vis[x][y][layer]=true; Q.push(Point(x,y,now.step+1,layer)); } } } return -1;
}
int main()
{ cin>>t; while(t--) { cin>>n>>m>>k;//输入行和列以及k值! for(int i=1; i<=n; i++) for(int j=1; j<=m; j++) cin>>map[i][j];//输入地图 cout<<ans()<<endl; } return 0;
}