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
资源约定:
峰值内存消耗(含虚拟机) < 512M
CPU消耗 < 2000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
求最短路径,用广度优先搜索
n = int(input()) #模拟地图 mMap = [list(input().split()) for _ in range(n)] #标记是否访问过、访问过为True未访问过为False mVis = [[False for _ in range(n)] for _ in range(n) ] mVis[0][0] = True #模拟在图中上下左右走 xMove = [-1,1,0,0] yMove = [0,0,1,-1] #队列记录坐标和走到的次数 queue = [[0,0,0]] while queue: #当走到了b点退出广搜、打印走的次数 if mMap[queue[0][0]][queue[0][1]] == 'B': print(queue[0][2]) break for i in range(4): dx = queue[0][0] + xMove[i] dy = queue[0][1] + yMove[i] if dx>=0 and dx<=n-1 and dy>=0 and dy<=n-1: if not mVis[dx][dy] and mMap[queue[0][0]][queue[0][1]] != mMap[dx][dy]: '''没有访问过并且不是连续的走-和+就可以将点加入队列''' queue.append([dx,dy,queue[0][2]+1]) mVis[dx][dy] = True queue.pop(0) if len(queue) == 0: """队列为空说明走不到B点""" print("-1")