网易笔试-数据分析岗
最后一个编程题，大概题目是在一个n*m的地图上，每个位置都有一个怪兽0或者一个英雄1。每个位置离上下左右相邻的距离都是1，整个地图至少有1个怪兽。如果当前位置是0，就输出0。求每个英雄距离最近的怪兽的距离。
输入：第一行输入两个正整数n和m，接下来每行输入m个整数0或者1。
输出：n行，每行m个用空格分隔的数字表示最近的距离，要求行末没有多余的空格
我感觉题目有歧义，就是没说算不算拐角，还是只跟上下左右比就行了。
比如：111，110，第一行第一个1的结果应该是0 呢还是3呢？
做了很久很久都没做出来，后来才做出来，而且当作只横向比和纵向比，不拐角。也不知道通过率能有多少，有没有大佬能指点一下？
def main(n,m,x,d_i,d_j): res=0 res1=0 res2=0 if x[d_i][d_j]=='0': #如果当前位置是怪兽，输出0 res=0 else: for k in range(d_j,m): if x[d_i][k]=='0': #找到同一行第一个怪兽 res1=k-d_j break for k in range(d_i,n): #找到同一列第一个怪兽 if x[k][d_j]=='0': res2=k-d_i break if res1!=0 and res2!=0: res=min(res1,res2) #看看哪个近 else: res=max(res1,res2) #如果有一行/列没有怪兽，就返回有数字的那个 return res if __name__=='__main__': n,m=2,3 x=['111','110'] for d_i in range(n): for d_j in range(m-1): print(main(n,m,x,d_i,d_j),end=' ') #d_i,d_j是为了定位当前位置 d_j=m-1 print(main(n,m,x,d_i,d_j)) #为了一行的最后没有空格
