暴力目前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)