蓝桥 穿越雷区【第六届】【决赛】【C组】 【 Python】 （BFS求最小步数）

题目

这里是引用问题描述：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

分析

求 最少移动步数 ，即找到最短路径就可以，典型的bfs题

本题的重中之重就是最少移动步数。给出了一个图形，让我们求从起点到终点的路径，常见的搜索算法有DFS，跟BFS。如果要求起点到终点的路径总和，我们使用DFS来求解比较方便些，**而如果要求起点到终点的最短路径，那无疑就是使用BFS了。**比如在下象棋，BFS就是你走一步我走一步，DFS就是一直是我走，等我吃完你的帅然后才到你走。

注意bfs的几个必备部分，queue=[], pre={}记录前一个节点，judge判断函数 ，dir方向数组；

1. 注意 倒序遍历pre的方法

                    while pre[(nx,ny)]!=st:#倒序输出就是最短路径
                        #print((nx,ny))
                        (nx,ny)=pre[(nx,ny)]
                        step+=1

同时记录最短路径

2. 注意这里的judge函数多了一个判断
3. 如何保证最短路径（与dfs的vis数组区分）

f (nx,ny) not in pre.keys():#保证最短路没有访问过，保证最短！

代码

n=int(input())
dp=[]
for i in range(n):
    t=input().split()
    if 'A' in t:
        st=(i,t.index('A'))
    if 'B' in t:
        end=(i,t.index('B'))
    dp.append(t)
#print(dp)
#print(st,end) 
pre={}#记录路径，使用字典，已经保证最短！
dir=[(1,0),(-1,0),(0,1),(0,-1)]

def judge(nx,ny,flag):
    if 0<=nx<=n-1 and 0<=ny<=n-1:
        if flag!=dp[nx][ny]:
            if (nx,ny) not in pre.keys():#保证最短路没有访问过，保证最短！
                return True
    return False
        
    
def bfs():
    queue=[st]
    step=0
    while queue:
        x,y=queue.pop(0)
        flag=dp[x][y]#获取当前坐标的物体
        #找到目标位置
        if (x,y) == end:
            step=1#默认值为1
            while pre[(x,y)]!=st:
                (x,y)=pre[(x,y)]
                step+=1
            print(step)
            break
        
        
        for i in dir:
            nx,ny=x+i[0],y+i[1]
            if judge(nx,ny,flag):
                queue.append((nx,ny))
                pre[(nx,ny)]=(x,y)#记录路径

    if step==0:   #如果没搜到B setp不会更新 
        print('-1')
            
    
bfs()

反思

本题数据量4<n<100 ,不是很大，所以queue=[]用列表模拟就够了，
但因为列表删除和插入的复杂度是O（n），如果数据量大n在1000，效率会很慢。
这个时候

需要借助PY自带的库里面的deque(双向队列)，它的时间复杂是O（1），以后也尽量用自带的，因为比赛时并不能看见结果，要最优化写！。
collections.deque学习传送门 Python collections模块之deque()详解