LeetCode 1030. Matrix Cells in Distance Order
考点 | 难度 |
---|---|
Sorting | Easy |
题目
You are given four integers row
, cols
, rCenter
, and cCenter
. There is a rows x cols matrix and you are on the cell with the coordinates (rCenter
, cCenter
).
Return the coordinates of all cells in the matrix, sorted by their distance from (rCenter
, cCenter
) from the smallest distance to the largest distance. You may return the answer in any order that satisfies this condition.
The distance between two cells (r1, c1)
and (r2, c2)
is |r1 - r2| + |c1 - c2|
.
思路
先把所有坐标存起来,再用comparator对它们进行排序。
答案
public int[][] allCellsDistOrder(int R, int C, int r, int c) {
int index = 0;
int[][] result = new int[R*C][2];
for(int i = 0; i < R; i++){
for(int j = 0; j < C; j++){
result[index][0] = i;
result[index][1] = j;
index++;
}
}
Arrays.sort(result, (a, b)->
(Math.abs(r-a[0])+Math.abs(c-a[1])) - (Math.abs(r-b[0])+Math.abs(c-b[1]))
);
return result;
}