蓝桥杯-途径最短街道数-dfs-bfs-python题解

蓝桥杯-途径最短街道数

小蓝住在 LQ 城，今天他要去小乔家玩。
LQ 城可以看成是一个 n 行 m 列的一个方格图。
小蓝家住在第 1 行第 1 列，小乔家住在第 n 行第 m 列。
小蓝可以在方格图内走，他不愿意走到方格图外。
城市中有的地方是风景优美的公园，有的地方是熙熙攘攘的街道。小蓝很喜欢公园，不喜欢街道。他把方格图中的每一格都标注了一个属性，或者是喜欢的公园，标为1，或者是不喜欢的街道标为2。小蓝和小乔住的地方都标为了1。
小蓝每次只能从一个方格走到同一行或同一列的相邻方格。他想找到一条路径，使得不连续走两次标为 2 的街道，请问在此前提下他最少要经过几次街道？
输入格式
输入的第一行包含两个整数 n, m，用一个空格分隔。
接下来 n 行，每行一个长度为 m 第数字串，表示城市的标注。
输出格式
输出一行包含一个整数，表示答案。如果没有满足条件的方案，输出 -1。
样例输入
3 4
1121
1211
2211
样例输出
1
样例输入
3 4
1122
1221
2211
样例输出
-1
样例输入
5 6
112121
122221
221212
211122
111121
样例输出
5
评测用例规模与约定
对于 50% 的评测用例，2 <= n, m <= 20。
对于所有评测用例，2 <= n, m <= 300。

优先选bfs，因为要考虑是否有解

#需要先判断是否有解再去找最优解，所以我们采用bfs
from queue import *

n,m=map(int,input().split())
s=[]
for i in range(n):
    l=list(input())
    s.append(l)

v=[[0 for i in range (m)]for j in range(n)]

ans=0x3f3f3f 
dx=[1,-1,0,0]
dy=[0,0,1,-1]


q=Queue()

def bfs(x,y):

    global ans
    q.put((x,y,0,1))
    v[x][y]=1
    while not q.empty():
        a=q.get()

        if a[0]==n-1 and a[1]==m-1:
            if a[2]<ans:
                ans=a[2]
            return
        for i in range(4):
            x1=a[0]+dx[i]
            y1=a[1]+dy[i]
            if x1<0 or x1>=n or y1<0 or y1>=m or v[x1][y1]==1:
                continue
            if s[x1][y1]=='2' and a[3]==0:
                continue
            if s[x1][y1]=='2' and a[3]==1:
                flag=0
                res1=a[2]+1
                q.put((x1,y1,res1,flag))
                v[x1][y1]=1

            if s[x1][y1]=='1':
                flag=1
                q.put((x1,y1,a[2],flag))
                v[x1][y1]=1

            
bfs(0,0)
if v[n-1][m-1]==1:
    print(ans)
else:
    print(-1)
            
        
    
    

若选dfs，则需设置搜索深度，找不到解就返回。

import sys
sys.setrecursionlimit(100000)
n,m=map(int,input().split())
s=[]
for i in range(n):
    l=list(input())
    s.append(l)

visit=[[0 for i in range (m)]for j in range(n)]

ans=0x3f3f3f 
dx=[1,-1,0,0]
dy=[0,0,1,-1]
flag=1
def dfs(x,y,res):
    global flag
    global ans
    if x==n-1 and y==m-1:
        if res<ans:
            ans=res
        return
    for i in range(4):
        x+=dx[i]
        y+=dy[i]
        if x>=0 and x<n and y>=0 and y<m and visit[x][y]==0:
            visit[x][y]=1
            if s[x][y]=='1' and flag==1:
                if res<ans:
                    dfs(x,y,res)
            if s[x][y]=='1' and flag==0:
                flag=1
                if res<ans:
                    dfs(x,y,res)
                flag=0

            if s[x][y]=='2' and flag==1:
                flag=0
                if res<ans:
                    dfs(x,y,res+1)
                flag=1
            visit[x][y]=0
        x-=dx[i]
        y-=dy[i]
        
                
dfs(0,0,0)
if ans==0x3f3f3f:
    print(-1)
else:
    print(ans)