BFS走一遍,将A与B分别放入两个队列中,每次将队列内的一层元素出队再进行下一层(将此步的可到达的地方全部都扩展到,保证两个队列同时扩展同一步。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <iomanip>
#include <queue>
using namespace std;
#define x first
#define y second
typedef pair<int, int> PII;
const int N = 1010;
int n, m;
char g[N][N];
bool st[2][N][N];
queue<PII> q[2];
int res;
int dx[8] = {-1, 0, 1, 0, -1, 1, 1, -1}, dy[8] = {0, 1, 0, -1, 1, 1, -1, -1};
void print(int u) {
for (int i = 0; i < n; i ++ ) {
for (int j = 0; j < m; j ++ )
cout << setw(5) << st[u][i][j];
cout << endl;
}
puts("====================================");
}
bool bfs(int u) {
int si = q[u].size();
//将队列内这一层都要出队
while (si --) {
PII t = q[u].front();
q[u].pop();
for (int i = 0; i < (u ? 4 : 8); i ++ ) {
int x = t.x + dx[i], y = t.y + dy[i];
if (x >= 0 && x < n && y >= 0 && y < m && !st[u][x][y] && g[x][y] != '#') {
if (st[u ^ 1][x][y]) return true;
st[u][x][y] = true;
q[u].push({x, y});
}
}
}
return 0;
}
int work() {
while (q[0].size() || q[1].size()) {
res ++;
if (bfs(0)) return res;
if (bfs(1)) return res;
if (bfs(1)) return res;
}
return 0;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> n >> m;
for (int i = 0; i < n; i ++ )
for (int j = 0; j < m; j ++ )
cin >> g[i][j];
for (int i = 0; i < n; i ++ )
for (int j = 0; j < m; j ++ )
if (g[i][j] == 'C') q[0].push({i, j}), st[0][i][j] = true;
else if (g[i][j] == 'D') q[1].push({i, j}), st[1][i][j] = true;
if (work()) cout << "YES\n" << res << endl;
else cout << "NO" << endl;
// print(0);
// print(1);
}

京公网安备 11010502036488号