暴力目前98%-----找数字

题目描述
给一个二维数组nums,对于每一个元素nums[i],找出距离最近的且值相等的元素,

输出横纵坐标差值的绝对值之和,如果没有等值元素,则输出-1。

例如:

输入数组 nums 为

0 3 5 4 2
2 5 7 8 3
2 5 4 2 4

输出为:

-1 4 2 3 3
1 1 -1 -1 4
1 1 2 3 2

对于 nums[0][0] = 0,不存在相等的值。
对于 nums[0][1] = 3,存在一个相等的值,最近的坐标为 nums[1][4],最小距离为 4。
对于 nums[0][2] = 5,存在两个相等的值,最近的坐标为 nums[1][1],故最小距离为 2。

对于 nums[1][1] = 5,存在两个相等的值,最近的坐标为 nums[2][1],故最小距离为1。

输入描述
输入第一行为二维数组的行

输入第二行为二维数组的列

输入的数字以空格隔开。

输出描述
数组形式返回所有坐标值。

备注
针对数组 nums[i][j],满足 0 < i ≤ 100,0 < j ≤ 100
对于每个数字,最多存在 100 个与其相等的数字
用例1
输入
3
5
0 3 5 4 2
2 5 7 8 3
2 5 4 2 4
输出
[[-1, 4, 2, 3, 3], [1, 1, -1, -1, 4], [1, 1, 2, 3, 2]]

n=int(input())
m=int(input())
matrix = [list(map(int,input().split())) for _ in range(n)]
#找一下相同数字的位置,并记录下来
nums={}
for i in range(n):
    for j in range(m):
        num = matrix[i][j]
        if nums.get(num) is None:
            nums[num] =[[i,j]]
        else:
            nums[num].append([i,j])

#便立即真,找到其相同数字位置最小的距离
for i in range(n):
    for j in range(m):
        num = matrix[i][j]
        mindis = 300  #距离肯定不可能过300,随便设一个得了
        for i1,j1 in nums[num]:
            if i1!=i or j1!=j:
                mindis=min(mindis,abs(i1-i)+abs(j1-j))
        matrix[i][j] = -1 if mindis==300 else mindis
print(matrix)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值