网易笔试-数据分析岗

最后一个编程题,大概题目是在一个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))   #为了一行的最后没有空格


#网易实习##网易##笔试题目#
全部评论
力扣原题有的
点赞 回复 分享
发布于 2020-04-07 22:57

相关推荐

ProMonkey2024:5个oc?厉害! 但是有一个小问题:谁问你了?😡我的意思是,谁在意?我告诉你,根本没人问你,在我们之中0人问了你,我把所有问你的人都请来 party 了,到场人数是0个人,誰问你了?WHO ASKED?谁问汝矣?誰があなたに聞きましたか?누가 물어봤어?我爬上了珠穆朗玛峰也没找到谁问你了,我刚刚潜入了世界上最大的射电望远镜也没开到那个问你的人的盒,在找到谁问你之前我连癌症的解药都发明了出来,我开了最大距离渲染也没找到谁问你了我活在这个被辐射蹂躏了多年的破碎世界的坟墓里目睹全球核战争把人类文明毁灭也没见到谁问你了(别的帖子偷来的,现学现卖😋)
点赞 评论 收藏
分享
评论
点赞
7
分享
牛客网
牛客企业服务