go語言LeetCode題解1030距離順序排列矩陣單元格

更新時間：2022年12月28日 09:41:23 作者：劉09k11

這篇文章主要為大家介紹了go語言LeetCode題解1030距離順序排列矩陣單元格，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪

一描述

1030. 距離順序排列矩陣單元格 - 力扣（LeetCode） (leetcode-cn.com)

給定四個整數(shù) row , cols , rCenter 和 cCenter 。有一個 rows x cols 的矩陣，你在單元格上的坐標(biāo)是 (rCenter, cCenter) 。

返回矩陣中的所有單元格的坐標(biāo)，并按與 (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) 周圍的單元格全部遍歷一遍

解題的關(guān)鍵在于定義好 less 函數(shù), 相當(dāng)于 Comparable 接口中的 compareTo 方法

步驟分為兩步

構(gòu)建一個二維數(shù)組

比較大小排序即可我這里使用快速排序

三答案

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距離順序排列矩陣單元格的詳細內(nèi)容，更多關(guān)于go 距離順序排列矩陣單元格的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: