【华为OD机试2023】找数字(Python)

题目描述：

给一个二维数组nums，对于每一个元素num[i]，找出距离最近的且值相等的元素，输出横纵坐标差值的绝对值之和，如果没有等值元素，则输出-1。 例如： 输入数组nums为  0 3 5 4 2 2 5 7 8 3 2 5 4 2 4

对于 num[0][0] = 0，不存在相等的值。 对于 num[0][1] = 3，存在一个相等的值，最近的坐标为num[1][4]，最小距离为4。 对于 num[0][2] = 5，存在两个相等的值，最近的坐标为num[1][1]，故最小距离为2。 ... 对于 num[1][1] = 5，存在两个相等的值，最近的坐标为num[2][1]，故最小距离为1。 ... 故输出为 -1 4  2  3 3  1 1 -1 -1 4  1 1  2  3 2

输入描述：

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

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

输入的数字以空格隔开。

输出描述：

数组形式返回所有坐标值。

补充说明：

1. 针对数组num[i][j]，满足0<i<=100；0<j<=100。

2. 对于每个数字，最多存在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 i in range(n)]
nums={}
for i in range(n):
    for j in range(m):
        num=matrix[i][j]
        if nums.get(num)==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]
        dis_min=None
        for i1,j1 in nums[num]:
            if i1!=i or j1!=j:
                dis=abs(i1-i)+abs(j1-j)
                if dis_min==None:
                    dis_min=dis
                else:
                    dis_min=min(dis,dis_min)
        matrix[i][j]=-1 if dis_min==None else dis_min

print(matrix)

二维数组还是不太行，学一下CSDN上的付费专栏答案