先看题目:https://ac.nowcoder.com/acm/problem/14608
题目描述:
迷宫问题升级版,有两种特殊格子'F','M',走了一种格子就不能再走另一种格子。问你最短步数?
解题思路:
初始化dis为-1,分两次bfs计算最短路,第一次bfs时'F'可走,第二次bfs时'M'可走。
两次bfs有很多需要注意与学习的,比如dis要重新memset、最后怎么输出解 等细节问题。
代码:
#include<bits/stdc++.h>
using namespace std;
#define INF 213323012
int n, m, r, c, T;
char mp[1010][1010];
char mp2[1010][1010];
int dir[4][2]={0,1,0,-1,1,0,-1,0};
int dis[1010][1010];
queue<int> q;
int bfs()
{
while(!q.empty()) q.pop();
q.push(0);
dis[0][0]=0;
while(!q.empty()) {
int tmp = q.front();
q.pop();
int x = tmp/m;
int y = tmp%m;
for(int i = 0; i < 4; i++) {
int dx = x + dir[i][0];
int dy = y + dir[i][1];
if(mp2[dx][dy] != '.' || dis[dx][dy] != -1 || dx < 0 || dx >= n || dy < 0 || dy >= m) continue;
dis[dx][dy] = dis[x][y]+1;
if(dx == r-1 && dy == c-1) return 1;
else q.push(dx*m+dy);
}
}
return 0;
}
int main()
{
scanf("%d",&T);
while(T--) {
memset(dis,-1, sizeof(dis));
scanf("%d%d%d%d",&n,&m,&r,&c);
for(int i = 0; i < n; i++)
for(int j = 0; j < m; j++) {
scanf(" %c",&mp[i][j]);
}
for(int i = 0; i < n; i++)
for(int j = 0; j < m; j++) {
if(mp[i][j] == 'M') mp2[i][j] = '.';
else mp2[i][j] = mp[i][j];
}
bfs();
int ans1 = dis[r-1][c-1];
memset(dis,-1, sizeof(dis));
for(int i = 0; i < n; i++)
for(int j = 0; j < m; j++) {
if(mp[i][j] == 'F') mp2[i][j] = '.';
else mp2[i][j] = mp[i][j];
}
bfs();
int ans2 = dis[r-1][c-1];
if(ans1 == -1 && ans2 == -1) printf("IMPOSSIBLE\n");
else{
if(ans1 == -1 || ans2 == -1)
printf("%d\n",max(ans1,ans2)*2);
else
printf("%d\n",min(ans1,ans2)*2);
}
}
}

京公网安备 11010502036488号