LeetCode33--距离顺序排列矩阵单元格和数组的相对排序

minastinis of king 2021-01-25 12:08:25

分类专栏： Leetcode刷题 # Leetcode经典好题

本文链接：https://blog.csdn.net/qq_40971025/article/details/113107285

版权

1.距离顺序排列矩阵单元格

//给出 R 行 C 列的矩阵，其中的单元格的整数坐标为 (r, c)，满足 0 <= r < R 且 0 <= c < C。 
//
// 另外，我们在该矩阵中给出了一个坐标为 (r0, c0) 的单元格。 
//
// 返回矩阵中的所有单元格的坐标，并按到 (r0, c0) 的距离从最小到最大的顺序排，其中，两单元格(r1, c1) 和 (r2, c2) 之间的距离是曼哈
//顿距离，|r1 - r2| + |c1 - c2|。（你可以按任何满足此条件的顺序返回答案。） 
//
// 
//
// 示例 1： 
//
// 输入：R = 1, C = 2, r0 = 0, c0 = 0
//输出：[[0,0],[0,1]]
//解释：从 (r0, c0) 到其他单元格的距离为：[0,1]
// 
//
// 示例 2： 
//
// 输入：R = 2, C = 2, r0 = 0, c0 = 1
//输出：[[0,1],[0,0],[1,1],[1,0]]
//解释：从 (r0, c0) 到其他单元格的距离为：[0,1,1,2]
//[[0,1],[1,1],[0,0],[1,0]] 也会被视作正确答案。
// 
//
// 示例 3： 
//
// 输入：R = 2, C = 3, r0 = 1, c0 = 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 <= R <= 100 
// 1 <= C <= 100 
// 0 <= r0 < R 
// 0 <= c0 < C 
// 
// Related Topics 排序

今天两个题是都不简单，都有一定的技巧性，这一题主要是学会自己定义一个比较器，直接比较各点距离定点距离的大小，然后就自动排序了。还有其他的方法，可以看官方题解：

https://leetcode-cn.com/problems/matrix-cells-in-distance-order/solution/si-chong-jie-fa-shu-zu-pai-xu-tong-pai-xu-bfs-mo-n/

class Solution {
        public int[][] allCellsDistOrder(int R, int C, int r0, int c0) {
            int[][] re = new int[R * C][2];
            for (int i = 0; i < R; i++) {
                for (int j = 0; j < C; j++) {
                    int t = i * C + j;
                    re[t][0] = i;
                    re[t][1] = j;
                }
            }
            Arrays.sort(re, (arr1, arr2) -> {
                int d1 = dist(arr1[0], arr1[1], r0, c0);
                int d2 = dist(arr2[0], arr2[1], r0, c0);
                return Integer.compare(d1, d2);
            });

            return re;
        }
        private int dist(int r1,int c1,int r2,int c2) {
            return Math.abs(r1 - r2) + Math.abs(c1 - c2);
        }
    }

2.数组的相对排序

//给你两个数组，arr1 和 arr2， 
//
// 
// arr2 中的元素各不相同 
// arr2 中的每个元素都出现在 arr1 中 
// 
//
// 对 arr1 中的元素进行排序，使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末
//尾。 
//
// 
//
// 示例： 
//
// 
//输入：arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
//输出：[2,2,2,1,4,3,3,9,6,7,19]
// 
//
// 
//
// 提示： 
//
// 
// 1 <= arr1.length, arr2.length <= 1000 
// 0 <= arr1[i], arr2[i] <= 1000 
// arr2 中的元素 arr2[i] 各不相同 
// arr2 中的每个元素 arr2[i] 都出现在 arr1 中 
// 
// Related Topics 排序 数组

这个也可以用自定义的比较器，也可以用计数排序的方式，其中后面的方式更加巧妙，那就是生成一个新的数组，用于存储arr1中出现的值及其出现的次数，这样再遍历arr2，当arr2中一个元素，先将其全部填充进arr1，然后知道arr2遍历完，最后再把剩下的元素填充进arr1。因为是数组，arr1中的值在新的数组中表示为下标的形式，所以相当于进行排序了。

自定义比较器：

public int[] relativeSortArray(int[] arr1, int[] arr2) {
        Map<Integer, Integer> map = new HashMap<>();
        List<Integer> list = new ArrayList<>();
        for(int num : arr1) list.add(num);
        for(int i = 0; i < arr2.length; i++) map.put(arr2[i], i);
        Collections.sort(list, (x, y) -> {
            //如果两个数都是在arr2中的，那么直接位置的从小到大进行排序
            //或者说如果有一个数在arr2中，那么这个数肯定比不再arr2中的数排在前面
            //如果有两个数都不在arr2中，那么直接按值的从小到大进行排序
            if(map.containsKey(x) || map.containsKey(y)) return map.getOrDefault(x, 1001) - map.getOrDefault(y, 1001);
            return x - y;
        });
        for(int i = 0; i < arr1.length; i++) arr1[i] = list.get(i);
        return arr1;
}

计数比较：

public int[] relativeSortArray(int[] arr1, int[] arr2) {
        int[] count = new int[1001];
        //将arr1中的数记录下来
        for (int num1:arr1) {
            count[num1]++;
        }
        //先安排arr2中的数
        int i = 0;
        for (int num2:arr2) {
            while(count[num2] > 0){
                arr1[i++] = num2;
                count[num2]--;
            }
        }
        //再排剩下的数
        for (int j = 0; j < count.length; j++) {
            while (count[j] > 0){
                arr1[i++] = j;
                count[j]--;
            }
        }
        return arr1;
    }

官方题解：

https://leetcode-cn.com/problems/relative-sort-array/solution/ming-que-bi-jiao-fang-shi-hou-xiang-zen-yao-pai-ji/