找数字
题目
给一个二维数组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
输入
输入第一行为二维数组的行
输入第二行为二维数组的列
输入的数字以空格隔开。
输出
数组形式返回所有坐标值。
备注
- 针对数组
num[i][j]
,满足 0<i≤100;0<j≤100。 - 对于每个数字,最多存在 100 个与其相等的数字。
示例一
输入
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]]
编码思路
- 使用
defaultdict
创建num_pos
字典,key
为矩阵中出现的数字,value