我们在输入时顺手记下来传送门的位置。如果这个传送门的左端点已经被存储,存入右端点,否则存入左端点。
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>a[i][j];
if(a[i][j]>='A'&&a[i][j]<='Z'){ // 存储传送门的位置,直接拿字母当下标
if(!l[a[i][j]].x){
l[a[i][j]]={i,j};
}else{
r[a[i][j]]={i,j};
}
}else if(a[i][j]=='@'){
B={i,j};
}else if(a[i][j]=='='){
E={i,j};
}
}
}
随后从起点开始 BFS。
若当前节点为传送门,我们找到与其对应的传送门走过去,节点不需要加入队列,并且不需要被标记(传送门可以多次使用)。
for(int i='A';i<='Z';i++){
if(l[i].x){ // 该传送门存在
if(p.x==l[i].x&&p.y==l[i].y){
dis[r[i].x][r[i].y]=dis[p.x][p.y];
p.x=r[i].x,p.y=r[i].y; // 更改位置
break;
}else if(p.x==r[i].x&&p.y==r[i].y){
dis[l[i].x][l[i].y]=dis[p.x][p.y];
p.x=l[i].x,p.y=l[i].y;
break;
}
}
}
接下来我们向“上下左右”四个方向走,如果该格子没有被走过且不是墙,我们就走过去。
for(int i=0;i<4;i++){
int dx=p.x+xyx[i][0],dy=p.y+xyx[i][1];
if(dx&&dy&&dx<=n&&dy<=m&&!vis[dx][dy]&&a[dx][dy]!='#'){
dis[dx][dy]=dis[p.x][p.y]+1; // 步数加 1
vis[dx][dy]=true;
q.push({dx,dy});
}
}
这种写法的好处是用不着写 01-bfs(要学习 01-bfs 自行 BDFS 谢谢)。
总之,这道题就几个坑点:
坑点
- 传送门可以重复走,因为传送门只是个中介点。
- 别写傻了忘记判断墙能不能走(我被坑死了)。
实现
#include<bits/stdc++.h>
using namespace std;
#define int long long
struct node{
int x,y;
}l[255],r[255],B,E;
int n,m,dis[305][305],xyx[5][3]={{-1,0},{1,0},{0,1},{0,-1}};
char a[305][305];
bool vis[305][305];
void bfs(int X,int Y){
queue<node>q;
vis[X][Y]=true;
q.push({X,Y});
dis[X][Y]=0;
while(q.size()){
auto p=q.front();
q.pop();
for(int i='A';i<='Z';i++){
if(l[i].x){
if(p.x==l[i].x&&p.y==l[i].y){
dis[r[i].x][r[i].y]=dis[p.x][p.y];
p.x=r[i].x,p.y=r[i].y;
break;
}else if(p.x==r[i].x&&p.y==r[i].y){
dis[l[i].x][l[i].y]=dis[p.x][p.y];
p.x=l[i].x,p.y=l[i].y;
break;
}
}
}
for(int i=0;i<4;i++){
int dx=p.x+xyx[i][0],dy=p.y+xyx[i][1];
if(dx&&dy&&dx<=n&&dy<=m&&!vis[dx][dy]&&a[dx][dy]!='#'){
dis[dx][dy]=dis[p.x][p.y]+1;
vis[dx][dy]=true;
q.push({dx,dy});
}
}
}
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>a[i][j];
if(a[i][j]>='A'&&a[i][j]<='Z'){
if(!l[a[i][j]].x){
l[a[i][j]]={i,j};
}else{
r[a[i][j]]={i,j};
}
}else if(a[i][j]=='@'){
B={i,j};
}else if(a[i][j]=='='){
E={i,j};
}
}
}
bfs(B.x,B.y);
cout<<dis[E.x][E.y];
return 0;
}