网易笔试-数据分析岗
最后一个编程题,大概题目是在一个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)) #为了一行的最后没有空格