【华为0920】【python】快速传球

练习

题目来源 公众号 万诺coding
题目内容
班级组织传球活动，男女同学随机排成m行n列队伍，第一列中的任意一个男同学都可以作为传球的起点，要求最终将球传到最后一列的任意一个男同学手里，求所有能够完成任务的传球路线中的最优路线(传球次数最少的路线)的传球次数。

def dfs(i,j,flag,pre):
    ans = m*n
    for (x,y) in ((i-1,j),(i+1,j),(i,j-1),(i,j+1)):
        if x<0 or x>=m or y<0 or y>=n or matrix[x][y]==0 or flag[x][y]==1:
            continue
        if y==n-1:
            return pre+1
        flag[x][y]=1
        ans = min(ans,dfs(x,y,flag,pre+1))
        flag[x][y]=0
    return ans

m=4
n=4
matrix = ['1 1 1 0','1 1 1 0','0 0 1 0','0 1 1 1']
matrix = [list(map(int,x.split(' '))) for x in matrix]

flag = [[0 for _ in range(n)] for _ in range(m)]
ans=m*n
pre = 0
for i in range(m):
    if matrix[i][0]==0:
        continue
    flag[i][0]=1
    ans = min(ans,dfs(i,0,flag,pre))
    flag[i][0]=0
print(ans)