go语言LeetCode题解1030距离顺序排列矩阵单元格
作者:刘09k11
这篇文章主要为大家介绍了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; } }
以上就是go语言LeetCode题解1030距离顺序排列矩阵单元格的详细内容,更多关于go 距离顺序排列矩阵单元格的资料请关注脚本之家其它相关文章!