距离顺序排列矩阵单元格(python)

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))
# [[0,2],[0,1],[1,2],[0,0],[1,1],[2,2],[1,0],[2,1],[2,0]]