go语言LeetCode题解1030距离顺序排列矩阵单元格

一 描述

1030. 距离顺序排列矩阵单元格 - 力扣（LeetCode） (leetcode-cn.com)

给定四个整数 row ,   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 <= 100

0 <= rCenter < rows

0 <= cCenter < cols

二 分析

这个问题其实用任何排序都可以做, 不用那么暴力还把 (r0,c0) 周围的单元格全部遍历一遍

解题的关键在于定义好 less 函数, 相当于 Comparable 接口中的 compareTo 方法

步骤分为两步

构建一个二维数组

比较大小排序即可 我这里使用快速排序

三 答案

class Solution {
    public  int[][] allCellsDistOrder(int R, int C, int r0, int c0) {
       int[][] a = new int[R * C][2];
       for (int i = 0; i < R; i++) {
           for (int j = 0; j < C; j++) {
               a[i * C + j] = new int[]{i, j};
           }
       }
       sort(a, 0, a.length - 1, r0, c0);
       return a;
   }
   private static void sort(int[][] a, int lo, int hi, int r0, int c0) {
       if (hi <= lo) {
           return;
       }
       int j = partition(a, lo, hi, r0, c0);
       sort(a, lo, j - 1, r0, c0);
       sort(a, j + 1, hi, r0, c0);
   }
private static int partition(int[][] a, int lo, int hi, int r0, int c0) {
       int i = lo;
       int j = hi + 1;
       while (true) {
           while (less(a, ++i, lo, r0, c0)) {
               if (i >= hi) {
                   break;
               }
           }
           while (less(a, lo, --j, r0, c0)) {
               if (j <= lo) {
                   break;
               }
           }
           if (i >= j) {
               break;
           }
           swap(a, i, j);
       }
       swap(a, lo, j);
       return j;
   }
   private static void swap(int[][] a, int i, int j) {
       int[] tmp = a[i];
       a[i] = a[j];
       a[j] = tmp;
   }
   private static boolean less(int[][] a, int i, int j, int ro, int c0) {
       int[] arr1 = a[i];
       int[] arr2 = a[j];
       int len1 = Math.abs(arr1[0] - ro) + Math.abs(arr1[1] - c0);
       int len2 = Math.abs(arr2[0] - ro) + Math.abs(arr2[1] - c0);
       return len1 - len2 < 0;
   }
}

来源：https://juejin.cn/post/7181771721396092987