1030.距离顺序排列矩阵单元格
给出R行C列的矩阵,其中的单元格的整数坐标为(r, c),满足0 <= r < R且0 <= c < C。
另外,我们在该矩阵中给出了一个坐标为(r0, c0)的单元格。
返回矩阵中的所有单元格的坐标,并按到(r0, c0)的距离从最小到最大的顺序排,其中,两单元格(r1, c1)
和(r2, c2)之间的距离是曼哈顿距离, | r1 - r2 | + | c1 - c2 |。(你可以按任何满足此条件的顺序返回答案。)
示例1:
输入:R = 1, C = 2, r0 = 0, c0 = 0
输出:[[0, 0], [0, 1]]
解释:从(r0, c0)到其他单元格的距离为:[0, 1]
示例2:
输入:R = 2, C = 2, r0 = 0, c0 = 1
输出:[[0, 1], [0, 0], [1, 1], [1, 0]]
解释:从(r0, c0)到其他单元格的距离为:[0, 1, 1, 2]
[[0, 1], [1, 1], [0, 0], [1, 0]]也会被视作正确答案。
示例3:
输入:R = 2, C = 3, r0 = 1, c0 = 2
输出:[[1, 2], [0, 2], [1, 1], [0, 1], [1, 0], [0, 0]]
解释:从(r0, c0)到其他单元格的距离为:[0, 1, 1, 2, 2, 3]
其他满足题目要求的答案也会被视为正确,例如[[1, 2], [1, 1], [0, 2], [1, 0], [0, 1], [0, 0]]。
def func(R, C, r0, c0):
dp = []
for i in range(R):
for j in range(C):
dp.append([i, j])
def sort_key(x):
return abs(x[0]-r0)+abs(x[1]-c0)
dp.sort(key=sort_key)
return dp
print(func(R = 3, C = 3, r0 = 0, c0 = 2))