class Solution {
public:
vector<vector<int>> allCellsDistOrder(int rows, int cols, int rCenter, int cCenter) {
}
};
1030. 距离顺序排列矩阵单元格
给定四个整数 rows , cols , rCenter 和 cCenter 。有一个 rows x cols 的矩阵，你在单元格上的坐标是 (rCenter, cCenter) 。
返回矩阵中的所有单元格的坐标，并按与 (rCenter, cCenter) 的 距离 从最小到最大的顺序排。你可以按 任何 满足此条件的顺序返回答案。
单元格(r1, c1) 和 (r2, c2) 之间的距离为|r1 - r2| + |c1 - c2|。
示例 1：
输入：rows = 1, cols = 2, rCenter = 0, cCenter = 0 输出：[[0,0],[0,1]] 解释：从 (r0, c0) 到其他单元格的距离为：[0,1]
示例 2：
输入：rows = 2, cols = 2, rCenter = 0, cCenter = 1 输出：[[0,1],[0,0],[1,1],[1,0]] 解释：从 (r0, c0) 到其他单元格的距离为：[0,1,1,2] [[0,1],[1,1],[0,0],[1,0]] 也会被视作正确答案。
示例 3：
输入：rows = 2, cols = 3, rCenter = 1, cCenter = 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]]。
提示：
1 <= rows, cols <= 1000 <= rCenter < rows0 <= cCenter < cols原站题解
golang 解法, 执行用时: 28 ms, 内存消耗: 7.5 MB, 提交时间: 2020-11-17 07:46:16
func allCellsDistOrder(R int, C int, r0 int, c0 int) [][]int {
var ans [][]int
for i := 0; i < R; i++ {
for j := 0; j < C; j++ {
ans = append(ans, []int{i, j})
}
}
sort.Slice(ans, func(v1, v2 int) bool {
return math.Abs(float64(ans[v1][0] - r0)) + math.Abs(float64(ans[v1][1]-c0))<math.Abs(float64(ans[v2][0] - r0)) + math.Abs(float64(ans[v2][1]-c0))
})
return ans
}
python3 解法, 执行用时: 152 ms, 内存消耗: 15.3 MB, 提交时间: 2020-11-17 07:37:54
class Solution:
def allCellsDistOrder(self, R: int, C: int, r0: int, c0: int) -> List[List[int]]:
ans = []
for i in range(R):
for j in range(C):
ans.append([i, j])
ans.sort(key=lambda x: abs(x[0]-r0) + abs(x[1]-c0))
return ans