LeetCode知识点总结 - 1030

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;
}