Matrix Cells in Distance Order

1
2
3
4
5
6
7
8

class Solution:
    def allCellsDistOrder(self, rows: int, cols: int, rCenter: int, cCenter: int) -> List[List[int]]:
        res = []
        for i in range(rows):
            for j in range(cols):
                res.append([i, j])
        res.sort(key=lambda x: abs(x[0] - rCenter) + abs(x[1] - cCenter))
        return res