穿越雷区（Python）

题目描述

X 星的坦克战车很奇怪，它必须交替地穿越正能量辐射区和负能量辐射区才能保持正常运转，否则将报废。

某坦克需要从 A 区到 B 区去（ A，B 区本身是安全区，没有正能量或负能量特征），怎样走才能路径最短？

已知的地图是一个方阵，上面用字母标出了 A，B 区，其它区都标了正号或负号分别表示正负能量辐射区。

例如：

A + - + -

- + - - +

- + + + -

+ - + - +

B + - + -

坦克车只能水平或垂直方向上移动到相邻的区。

输入描述

第一行是一个整数 n，表示方阵的大小， 4≤n<100。

接下来是 n 行，每行有 n 个数据，可能是 A，B，+，- 中的某一个，中间用空格分开。A，B 都只出现一次。

输出描述

输出一个整数，表示坦克从 A 区到 B 区的最少移动步数。

如果没有方案，则输出 -1。

输入输出样例

示例

输入

5
A + - + -
- + - - +
- + + + -
+ - + - +
B + - + -

输出

10

运行限制

• 最大运行时间：1s
• 最大运行内存: 512M

def pd(x,y):#判断是否越界，是否被走过
  if 0<=x<n and 0<=y<n and vis[x][y]==0:return True
  return False
def bfs():
  global m,vis
  d=[(1,0),(-1,0),(0,1),(0,-1)]#控制方向
  while m:
    x,y,s,bs=m.pop(0)
    if x==ddx and y==ddy:return bs
    for i in range(4):#向周围移动
      dx,dy=x+d[i][0],y+d[i][1]
      if pd(dx,dy) and q[dx][dy]!=q[x][y]:#没越界且与正负交替
        vis[dx][dy]=1;m.append((dx,dy,q[dx][dy],bs+1))
n=int(input())
m=[];vis=[[0]*n for i in range(n)];q=[]#存放坐标；对走过的坐标进行记录；存放地图
for i in range(n):
  s=input().split()
  q.append(s)#存放地图
  for j in range(n):
    if s[j]=='A':m.append((i,j,'A',0));vis[i][j]=1#对A点进行标记
    if s[j]=='B':ddx,ddy=i,j
f=bfs()
if f>n*n:print(-1)
else:print(f)