题目
题解
BFS模板题。
就在模板题的基础上稍微加了点限制而已。
我用的是pair存的,first表示位置,将二维压缩成一维了;second表示步数;
要求正负交替,我就采用异或的思想进行,判断下一个位置能否作为落脚点即可。
不懂如何实现的二维压缩成一维的同学,评论区问一下,我会回复。
代码
#include<bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
const int N = 110;
int dx[] = {-1, 1, 0, 0};
int dy[] = {0, 0, -1, 1};
int n, Ax, Ay, Bx, By, a[N][N], vis[N][N];
char ch;
queue<PII> q;
int main()
{
cin>>n;
for(int i = 0;i < n;i ++)
for(int j = 0;j < n;j ++) {
cin>>ch, getchar();
if(ch == 'A') Ax = i, Ay = j;
else if(ch == 'B') Bx = i, By = j;
else if(ch == '-') a[i][j] = 0;
else if(ch == '+') a[i][j] = 1;
}
q.push(PII(Ax*n+By, 0));
vis[Ax][Ay] = 1;
while(!q.empty()) {
PII p = q.front();
q.pop();
int x = p.first / n, y = p.first % n;
for(int i = 0;i < 4;i ++) {
int tx = x + dx[i], ty = y + dy[i];
if(tx < 0 || ty < 0 || tx >= n || ty >= n) continue;
if(tx == Bx && ty == By) {cout << p.second + 1 << endl; return 0;}
if(vis[tx][ty]) continue;
if((x==Ax && y==Ay) || a[tx][ty] ^ a[x][y])
q.push(PII(tx*n+ty, p.second + 1)), vis[tx][ty] = 1;
}
}
return 0;
}