先看题目: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); } } }